def test_delete_model(self, mocker):
mock_method = mocker.patch.object(MlflowClient, "delete_model_version")
client = ModelsClient()
name = "simple-nn-model"
client.delete_model(name, '1')
mock_method.assert_called_once_with(name="simple-nn-model",
version="1")
def test_delete_model(self):
client = ModelsClient()
name = "simple-nn-model"
client.delete_model(name, '1')