File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -127,8 +127,8 @@ def run_pdp_inference(
127127 """Triggers PDP inference Databricks run."""
128128 if (
129129 not req .filepath_to_type
130- or not check_types (req .filepath_to_type .values (), SchemaType .PDP_COURSE )
131- or not check_types (req .filepath_to_type .values (), SchemaType .PDP_COHORT )
130+ or not check_types (req .filepath_to_type .values (), SchemaType .COURSE )
131+ or not check_types (req .filepath_to_type .values (), SchemaType .STUDENT )
132132 ):
133133 raise ValueError (
134134 "run_pdp_inference() requires PDP_COURSE and PDP_COHORT type files to run."
@@ -147,10 +147,10 @@ def run_pdp_inference(
147147 job_id ,
148148 job_parameters = {
149149 "cohort_file_name" : get_filepath_of_filetype (
150- req .filepath_to_type , SchemaType .PDP_COHORT
150+ req .filepath_to_type , SchemaType .STUDENT
151151 ),
152152 "course_file_name" : get_filepath_of_filetype (
153- req .filepath_to_type , SchemaType .PDP_COURSE
153+ req .filepath_to_type , SchemaType .COURSE
154154 ),
155155 "databricks_institution_name" : db_inst_name ,
156156 "DB_workspace" : databricks_vars [
You can’t perform that action at this time.
0 commit comments