update docs and test for predictor

Author: Chuyang Deng

src/sagemaker/predictor.py

@@ -113,7 +113,8 @@ def _delete_endpoint_config(self):
         self.sagemaker_session.delete_endpoint_config(self._endpoint_config_name)
 
     def delete_endpoint(self, delete_endpoint_config=True):
-        """Delete the Amazon SageMaker endpoint and endpoint configuration backing this predictor.
+        """Delete the Amazon SageMaker endpoint backing this predictor. Also delete the endpoint configuration attached
+        to it if delete_endpoint_config is True.
 
         Args:
             delete_endpoint_config (bool, optional): Flag to indicate whether to delete endpoint configuration together

tests/unit/test_predictor.py

@@ -389,6 +389,8 @@ def json_sagemaker_session():
     ims = Mock(name='sagemaker_session')
     ims.default_bucket = Mock(name='default_bucket', return_value=BUCKET_NAME)
     ims.sagemaker_runtime_client = Mock(name='sagemaker_runtime')
+    ims.sagemaker_client.describe_endpoint = Mock(return_value=ENDPOINT_DESC)
+    ims.sagemaker_client.describe_endpoint_config = Mock(return_value=ENDPOINT_CONFIG_DESC)
 
     ims.sagemaker_client.describe_endpoint = Mock(return_value=ENDPOINT_DESC)
     ims.sagemaker_client.describe_endpoint_config = Mock(return_value=ENDPOINT_CONFIG_DESC)
@@ -430,6 +432,8 @@ def ret_csv_sagemaker_session():
     ims = Mock(name='sagemaker_session')
     ims.default_bucket = Mock(name='default_bucket', return_value=BUCKET_NAME)
     ims.sagemaker_runtime_client = Mock(name='sagemaker_runtime')
+    ims.sagemaker_client.describe_endpoint = Mock(return_value=ENDPOINT_DESC)
+    ims.sagemaker_client.describe_endpoint_config = Mock(return_value=ENDPOINT_CONFIG_DESC)
 
     ims.sagemaker_client.describe_endpoint = Mock(return_value=ENDPOINT_DESC)
     ims.sagemaker_client.describe_endpoint_config = Mock(return_value=ENDPOINT_CONFIG_DESC)