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