Skip to content

Commit 4684fbb

Browse files
committed
Fix session
1 parent 61d2643 commit 4684fbb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_tuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def test_attach_tuning_pytorch(sagemaker_session):
342342
time.sleep(15)
343343
tuner.wait()
344344

345-
attached_tuner = HyperparameterTuner.attach(tuning_job_name)
345+
attached_tuner = HyperparameterTuner.attach(tuning_job_name, sagemaker_session=sagemaker_session)
346346
best_training_job = tuner.best_training_job()
347347
with timeout_and_delete_endpoint_by_name(best_training_job, sagemaker_session, minutes=20):
348348
predictor = attached_tuner.deploy(1, 'ml.c4.xlarge')

0 commit comments

Comments
 (0)