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):
117
117
model_trainer = ModelTrainer (
118
118
sagemaker_session = modules_sagemaker_session ,
119
119
training_image = DEFAULT_CPU_IMAGE ,
120
- hyperparameters_file = f"{ source_dir } /hyperparameters.json" ,
120
+ hyperparameters = f"{ source_dir } /hyperparameters.json" ,
121
121
source_code = source_code ,
122
122
base_job_name = "hp-contract-hyperparameter-json" ,
123
123
)
@@ -134,7 +134,7 @@ def test_hp_contract_hyperparameter_yaml(modules_sagemaker_session):
134
134
model_trainer = ModelTrainer (
135
135
sagemaker_session = modules_sagemaker_session ,
136
136
training_image = DEFAULT_CPU_IMAGE ,
137
- hyperparameters_file = f"{ source_dir } /hyperparameters.yaml" ,
137
+ hyperparameters = f"{ source_dir } /hyperparameters.yaml" ,
138
138
source_code = source_code ,
139
139
base_job_name = "hp-contract-hyperparameter-yaml" ,
140
140
)
You can’t perform that action at this time.
0 commit comments