Skip to content

Commit 4e5909b

Browse files
committed
update unit test to use name from training job
1 parent 4434f0a commit 4e5909b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/test_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1434,7 +1434,7 @@ def test_attach_no_logs(sagemaker_session, training_job_description):
14341434
def test_logs(sagemaker_session, training_job_description):
14351435
estimator = Estimator.attach(training_job_name="job", sagemaker_session=sagemaker_session)
14361436
estimator.logs()
1437-
sagemaker_session.logs_for_job.assert_called_with(estimator.latest_training_job, wait=True)
1437+
sagemaker_session.logs_for_job.assert_called_with(estimator.latest_training_job.name, wait=True)
14381438

14391439

14401440
def test_attach_without_hyperparameters(sagemaker_session, training_job_description):

0 commit comments

Comments
 (0)