@@ -171,17 +171,6 @@ def test_delete_model(LocalSession):
171
171
assert MODEL_NAME not in sagemaker .local .local_session .LocalSagemakerClient ._models
172
172
173
173
174
- @patch ('sagemaker.local.local_session.LocalSession' )
175
- def test_delete_model (LocalSession ):
176
- local_sagemaker_client = sagemaker .local .local_session .LocalSagemakerClient ()
177
-
178
- local_sagemaker_client .create_model (MODEL_NAME , PRIMARY_CONTAINER )
179
- assert MODEL_NAME in sagemaker .local .local_session .LocalSagemakerClient ._models
180
-
181
- local_sagemaker_client .delete_model (MODEL_NAME )
182
- assert MODEL_NAME not in sagemaker .local .local_session .LocalSagemakerClient ._models
183
-
184
-
185
174
@patch ('sagemaker.local.local_session.LocalSession' )
186
175
def test_describe_model (LocalSession ):
187
176
local_sagemaker_client = sagemaker .local .local_session .LocalSagemakerClient ()
@@ -252,17 +241,6 @@ def test_delete_endpoint_config(LocalSession):
252
241
assert ENDPOINT_CONFIG_NAME not in sagemaker .local .local_session .LocalSagemakerClient ._endpoint_configs
253
242
254
243
255
- @patch ('sagemaker.local.local_session.LocalSession' )
256
- def test_delete_endpoint_config (LocalSession ):
257
- local_sagemaker_client = sagemaker .local .local_session .LocalSagemakerClient ()
258
-
259
- local_sagemaker_client .create_endpoint_config (ENDPOINT_CONFIG_NAME , PRODUCTION_VARIANTS )
260
- assert ENDPOINT_CONFIG_NAME in sagemaker .local .local_session .LocalSagemakerClient ._endpoint_configs
261
-
262
- local_sagemaker_client .delete_endpoint_config (ENDPOINT_CONFIG_NAME )
263
- assert ENDPOINT_CONFIG_NAME not in sagemaker .local .local_session .LocalSagemakerClient ._endpoint_configs
264
-
265
-
266
244
@patch ('sagemaker.local.image._SageMakerContainer.serve' )
267
245
@patch ('sagemaker.local.local_session.LocalSession' )
268
246
@patch ('urllib3.PoolManager.request' )
0 commit comments