Address comments.

Author: Chuyang Deng

tests/integ/test_transformer.py

@@ -106,46 +106,6 @@ def test_attach_transform_kmeans(sagemaker_session):
         attached_transformer.wait()
 
 
-# def test_transformer_delete_model(sagemaker_session):
-#     data_path = os.path.join(DATA_DIR, 'one_p_mnist')
-#     pickle_args = {} if sys.version_info.major == 2 else {'encoding': 'latin1'}
-#
-#     train_set_path = os.path.join(data_path, 'mnist.pkl.gz')
-#     with gzip.open(train_set_path, 'rb') as f:
-#         train_set, _, _ = pickle.load(f, **pickle_args)
-#
-#     kmeans = KMeans(role='SageMakerRole', train_instance_count=1,
-#                     train_instance_type='ml.c4.xlarge', k=10, sagemaker_session=sagemaker_session,
-#                     output_path='s3://{}/'.format(sagemaker_session.default_bucket()))
-#
-#     kmeans.init_method = 'random'
-#     kmeans.max_iterations = 1
-#     kmeans.tol = 1
-#     kmeans.num_trials = 1
-#     kmeans.local_init_method = 'kmeans++'
-#     kmeans.half_life_time_size = 1
-#     kmeans.epochs = 1
-#
-#     records = kmeans.record_set(train_set[0][:100])
-#     with timeout(minutes=TRAINING_DEFAULT_TIMEOUT_MINUTES):
-#         kmeans.fit(records)
-#
-#     transform_input_path = os.path.join(data_path, 'transform_input.csv')
-#     transform_input_key_prefix = 'integ-test-data/one_p_mnist/transform'
-#     transform_input = kmeans.sagemaker_session.upload_data(path=transform_input_path,
-#                                                            key_prefix=transform_input_key_prefix)
-#
-#     transformer = _create_transformer_and_transform_job(kmeans, transform_input)
-#     with timeout(minutes=TRANSFORM_DEFAULT_TIMEOUT_MINUTES):
-#         transformer.wait()
-#
-#     transformer.delete_model()
-#
-#     with pytest.raises(Exception) as exception:
-#         sagemaker_session.sagemaker_client.describe_model(ModelName=transformer.model_name)
-#         assert 'Could not find model' in exception.value.message
-
-
 def test_transform_mxnet_vpc(sagemaker_session, mxnet_full_version):
     data_path = os.path.join(DATA_DIR, 'mxnet_mnist')
     script_path = os.path.join(data_path, 'mnist.py')

tests/unit/test_predictor.py

@@ -459,16 +459,6 @@ def test_delete_endpoint_with_config():
     sagemaker_session.delete_endpoint_config.assert_called_with('endpoint-config')
 
 
-def test_delete_non_existing_endpoint():
-    sagemaker_session = empty_sagemaker_session()
-    expected_error_message = 'The endpoint this config attached to does not exist.'
-    sagemaker_session.sagemaker_client.describe_endpoint = Mock(side_effect=ValueError(expected_error_message))
-    predictor = RealTimePredictor(ENDPOINT, sagemaker_session=sagemaker_session)
-
-    with pytest.raises(ValueError, match=expected_error_message):
-        predictor.delete_endpoint()
-
-
 def test_delete_endpoint_only():
     sagemaker_session = empty_sagemaker_session()
     predictor = RealTimePredictor(ENDPOINT, sagemaker_session=sagemaker_session)