Fix parameter name for wait

Author: Dan Choi

src/sagemaker/tuner.py

@@ -393,7 +393,7 @@ def start_new(cls, tuner, inputs):
         return cls(tuner.sagemaker_session, tuner._current_job_name)
 
     def stop(self):
-        self.sagemaker_session.stop_tuning_job(HyperParameterTuningJobName=self.name)
+        self.sagemaker_session.stop_tuning_job(name=self.name)
 
     def wait(self):
         self.sagemaker_session.wait_for_tuning_job(self.name)

tests/unit/test_tuner.py

@@ -331,7 +331,7 @@ def test_stop_tuning_job(sagemaker_session, tuner):
 
     tuner.stop_tuning_job()
 
-    sagemaker_session.stop_tuning_job.assert_called_once_with(HyperParameterTuningJobName=JOB_NAME)
+    sagemaker_session.stop_tuning_job.assert_called_once_with(name=JOB_NAME)
 
 
 def test_stop_tuning_job_no_tuning_job(tuner):
@@ -536,7 +536,7 @@ def test_stop(sagemaker_session):
     tuning_job = _TuningJob(sagemaker_session, JOB_NAME)
     tuning_job.stop()
 
-    sagemaker_session.stop_tuning_job.assert_called_once_with(HyperParameterTuningJobName=JOB_NAME)
+    sagemaker_session.stop_tuning_job.assert_called_once_with(name=JOB_NAME)
 
 
 def test_tuning_job_wait(sagemaker_session):