File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/integ/sagemaker/modules/train Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -117,7 +117,7 @@ def test_hp_contract_hyperparameter_json(modules_sagemaker_session):
117117 model_trainer = ModelTrainer (
118118 sagemaker_session = modules_sagemaker_session ,
119119 training_image = DEFAULT_CPU_IMAGE ,
120- hyperparameters_file = f"{ source_dir } /hyperparameters.json" ,
120+ hyperparameters = f"{ source_dir } /hyperparameters.json" ,
121121 source_code = source_code ,
122122 base_job_name = "hp-contract-hyperparameter-json" ,
123123 )
@@ -134,7 +134,7 @@ def test_hp_contract_hyperparameter_yaml(modules_sagemaker_session):
134134 model_trainer = ModelTrainer (
135135 sagemaker_session = modules_sagemaker_session ,
136136 training_image = DEFAULT_CPU_IMAGE ,
137- hyperparameters_file = f"{ source_dir } /hyperparameters.yaml" ,
137+ hyperparameters = f"{ source_dir } /hyperparameters.yaml" ,
138138 source_code = source_code ,
139139 base_job_name = "hp-contract-hyperparameter-yaml" ,
140140 )
You can’t perform that action at this time.
0 commit comments