We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 584980c commit 9056988Copy full SHA for 9056988
tests/unit/test_model.py
@@ -347,9 +347,7 @@ def test_model_delete_model(sagemaker_session, tmpdir):
347
sagemaker_session.delete_model.assert_called_with(model.name)
348
349
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))
+def test_delete_non_deployed_model(sagemaker_session):
+ model = DummyFrameworkModel(sagemaker_session)
354
with pytest.raises(ValueError, match='The SageMaker model must be deployed first before attempting to delete.'):
355
model.delete_model()
0 commit comments