Skip to content

Commit 9056988

Browse files
author
Chuyang Deng
committed
Remove unused parameters.
1 parent 584980c commit 9056988

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/unit/test_model.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -347,9 +347,7 @@ def test_model_delete_model(sagemaker_session, tmpdir):
347347
sagemaker_session.delete_model.assert_called_with(model.name)
348348

349349

350-
@patch('sagemaker.fw_utils.tar_and_upload_dir', MagicMock())
351-
@patch('time.strftime', MagicMock(return_value=TIMESTAMP))
352-
def test_delete_non_deployed_model(sagemaker_session, tmpdir):
353-
model = DummyFrameworkModel(sagemaker_session, source_dir=str(tmpdir))
350+
def test_delete_non_deployed_model(sagemaker_session):
351+
model = DummyFrameworkModel(sagemaker_session)
354352
with pytest.raises(ValueError, match='The SageMaker model must be deployed first before attempting to delete.'):
355353
model.delete_model()

0 commit comments

Comments
 (0)