Python ModelsClient.delete_model Examples

Example #1

File: test_model.py Project: ByronHsu/submarine

 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")

Example #2

 def test_delete_model(self):
     client = ModelsClient()
     name = "simple-nn-model"
     client.delete_model(name, '1')