def test_delete_registered_model(mock_get_request_message,
mock_model_registry_store):
name = "model_1"
mock_get_request_message.return_value = DeleteRegisteredModel(name=name)
_delete_registered_model()
_, args = mock_model_registry_store.delete_registered_model.call_args
assert args == {"name": name}
def test_delete_registered_model(mock_get_request_message,
mock_model_registry_store):
rm = RegisteredModel("model_1")
mock_get_request_message.return_value = DeleteRegisteredModel(
registered_model=rm.to_proto())
_delete_registered_model()
args, _ = mock_model_registry_store.delete_registered_model.call_args
assert args == (rm, )
def delete_registered_model(self, name):
"""
Delete the registered model.
Backend raises exception if a registered model with given name does not exist.
:param name: Registered model name.
:return: None
"""
req_body = message_to_json(DeleteRegisteredModel(name=name))
self._call_endpoint(DeleteRegisteredModel, req_body)
def delete_registered_model(self, name):
"""
Delete registered model.
Backend raises exception if a registered model with given name does not exist.
:param registered_model: :py:class:`mlflow.entities.model_registry.RegisteredModel` object.
:return: None
"""
req_body = message_to_json(DeleteRegisteredModel(name=name))
self._call_endpoint(DeleteRegisteredModel, req_body)
def _delete_registered_model():
request_message = _get_request_message(DeleteRegisteredModel())
_get_model_registry_store().delete_registered_model(
name=request_message.name)
return _wrap_response(DeleteRegisteredModel.Response())
def test_delete_registered_model(self, mock_http):
name = "model_1"
self.store.delete_registered_model(name=name)
self._verify_requests(mock_http, "registered-models/delete", "DELETE",
DeleteRegisteredModel(name=name))
def _delete_registered_model():
request_message = _get_request_message(DeleteRegisteredModel())
_get_model_registry_store().delete_registered_model(
RegisteredModel.from_proto(request_message.registered_model))
return _wrap_response(DeleteRegisteredModel.Response())
def test_delete_registered_model(self, mock_http):
rm = RegisteredModel("model_1")
self.store.delete_registered_model(registered_model=rm)
self._verify_requests(
mock_http, "registered-models/delete", "DELETE",
DeleteRegisteredModel(registered_model=rm.to_proto()))
