Skip to content

Commit a5dba4b

Browse files
committed
change: add integs
1 parent f8cc961 commit a5dba4b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/sagemaker/modules/train/test_model_trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)