Skip to content

Commit b95710d

Browse files
chuyang-dengChuyang Deng
authored andcommitted
Resolve flake8 error.
1 parent 05223e2 commit b95710d

File tree

1 file changed

+0
-22
lines changed

1 file changed

+0
-22
lines changed

tests/unit/test_local_session.py

Lines changed: 0 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -171,17 +171,6 @@ def test_delete_model(LocalSession):
171171
assert MODEL_NAME not in sagemaker.local.local_session.LocalSagemakerClient._models
172172

173173

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-
185174
@patch('sagemaker.local.local_session.LocalSession')
186175
def test_describe_model(LocalSession):
187176
local_sagemaker_client = sagemaker.local.local_session.LocalSagemakerClient()
@@ -252,17 +241,6 @@ def test_delete_endpoint_config(LocalSession):
252241
assert ENDPOINT_CONFIG_NAME not in sagemaker.local.local_session.LocalSagemakerClient._endpoint_configs
253242

254243

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-
266244
@patch('sagemaker.local.image._SageMakerContainer.serve')
267245
@patch('sagemaker.local.local_session.LocalSession')
268246
@patch('urllib3.PoolManager.request')

0 commit comments

Comments
 (0)