We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 61d2643 commit 4684fbbCopy full SHA for 4684fbb
tests/integ/test_tuner.py
@@ -342,7 +342,7 @@ def test_attach_tuning_pytorch(sagemaker_session):
342
time.sleep(15)
343
tuner.wait()
344
345
- attached_tuner = HyperparameterTuner.attach(tuning_job_name)
+ attached_tuner = HyperparameterTuner.attach(tuning_job_name, sagemaker_session=sagemaker_session)
346
best_training_job = tuner.best_training_job()
347
with timeout_and_delete_endpoint_by_name(best_training_job, sagemaker_session, minutes=20):
348
predictor = attached_tuner.deploy(1, 'ml.c4.xlarge')
0 commit comments