Add delete_endpoint() method for tuner (aws#48)

Author: laurenyu

src/sagemaker/tuner.py

@@ -208,6 +208,18 @@ def best_training_job(self):
         except KeyError:
             raise Exception('Best training job not available for tuning job: {}'.format(self.latest_tuning_job.name))
 
+    def delete_endpoint(self, endpoint_name=None):
+        """Delete an Amazon SageMaker endpoint.
+
+        If an endpoint name is not specified, this defaults to looking for an endpoint that
+        shares a name with the best training job for deletion.
+
+        Args:
+            endpoint_name (str): Name of the endpoint to delete
+        """
+        endpoint_name = endpoint_name or self.best_training_job()
+        self.sagemaker_session.delete_endpoint(endpoint_name)
+
     def _ensure_last_tuning_job(self):
         if self.latest_tuning_job is None:
             raise ValueError('No tuning job available')

tests/unit/test_tuner.py

@@ -432,6 +432,17 @@ def test_wait(tuner):
     tuner.estimator.sagemaker_session.wait_for_tuning_job.assert_called_once_with(JOB_NAME)
 
 
+def test_delete_endpoint(tuner):
+    tuner.latest_tuning_job = _TuningJob(tuner.estimator.sagemaker_session, JOB_NAME)
+
+    tuning_job_description = {'BestTrainingJob': {'TrainingJobName': JOB_NAME}}
+    tuner.estimator.sagemaker_session.sagemaker_client.describe_hyper_parameter_tuning_job = Mock(
+        name='describe_hyper_parameter_tuning_job', return_value=tuning_job_description)
+
+    tuner.delete_endpoint()
+    tuner.sagemaker_session.delete_endpoint.assert_called_with(JOB_NAME)
+
+
 #################################################################################
 # _ParameterRange Tests