Skip to content

Commit 2f9b45b

Browse files
chuyang-dengEliza Zhang
authored andcommitted
fix: change model_dir to training job name if it is for tuning. (aws#179)
* Change model_dir to training job if the training job is for tuning.
1 parent 80fddef commit 2f9b45b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/unit/test_training.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,4 +295,4 @@ def test_main_tuning_mpi_model_dir(configure_s3_env, read_hyperparameters, train
295295
training_env.return_value = single_machine_training_env
296296
os.environ['SAGEMAKER_REGION'] = REGION
297297
training.main()
298-
configure_s3_env.assert_called_once_with('/opt/ml/model', REGION)
298+
configure_s3_env.assert_called_once_with('/opt/ml/model', REGION)

0 commit comments

Comments
 (0)