def test_delete_model_version(mock_get_request_message, mock_model_registry_store):
name = "model1"
version = "32"
mock_get_request_message.return_value = DeleteModelVersion(name=name, version=version)
_delete_model_version()
_, args = mock_model_registry_store.delete_model_version.call_args
assert args == {"name": name, "version": version}
def test_delete_model_version(mock_get_request_message, mock_model_registry_store):
rm = RegisteredModel("model1")
mv = ModelVersion(registered_model=rm, version=32)
mock_get_request_message.return_value = DeleteModelVersion(model_version=mv.to_proto())
_delete_model_version()
args, _ = mock_model_registry_store.delete_model_version.call_args
assert args == (mv, )
def test_delete_model_version(self, mock_http):
name = "model_1"
version = "12"
self.store.delete_model_version(name=name, version=version)
self._verify_requests(
mock_http,
"model-versions/delete",
"DELETE",
DeleteModelVersion(name=name, version=version),
)
def delete_model_version(self, name, version):
"""
Delete model version in backend.
:param name: Registered model name.
:param version: Registered model version.
:return: None
"""
req_body = message_to_json(DeleteModelVersion(name=name, version=str(version)))
self._call_endpoint(DeleteModelVersion, req_body)
def delete_model_version(self, model_version):
"""
Delete model version in backend.
:param model_version: :py:class:`mlflow.entities.model_registry.ModelVersion` object.
:return: None
"""
req_body = message_to_json(DeleteModelVersion(model_version=model_version.to_proto()))
self._call_endpoint(DeleteModelVersion, req_body)
def _delete_model_version():
request_message = _get_request_message(DeleteModelVersion())
_get_model_registry_store().delete_model_version(
name=request_message.name, version=request_message.version)
return _wrap_response(DeleteModelVersion.Response())
def _delete_model_version():
request_message = _get_request_message(DeleteModelVersion())
_get_model_registry_store().delete_model_version(
ModelVersion.from_proto(request_message.model_version))
return _wrap_response(DeleteModelVersion.Response())
def test_delete_model_version(self, mock_http):
rm = RegisteredModel("model_1")
mv = ModelVersion(rm, 12)
self.store.delete_model_version(model_version=mv)
self._verify_requests(mock_http, "model-versions/delete", "DELETE",
DeleteModelVersion(model_version=mv.to_proto()))
