def _get_model_version_download_uri():
request_message = _get_request_message(GetModelVersionDownloadUri())
download_uri = _get_model_registry_store().get_model_version_download_uri(
ModelVersion.from_proto(request_message.model_version))
response_message = GetModelVersionDownloadUri.Response(
artifact_uri=download_uri)
return _wrap_response(response_message)
def _get_model_version_download_uri():
request_message = _get_request_message(GetModelVersionDownloadUri())
download_uri = _get_model_registry_store().get_model_version_download_uri(
name=request_message.name, version=request_message.version)
response_message = GetModelVersionDownloadUri.Response(
artifact_uri=download_uri)
return _wrap_response(response_message)
def test_get_model_version_download_uri(self, mock_http):
name = "model_11"
version = "8"
self.store.get_model_version_download_uri(name=name, version=version)
self._verify_requests(
mock_http, "model-versions/get-download-uri", "GET",
GetModelVersionDownloadUri(name=name, version=version))
def test_get_model_version_download_uri(self, mock_http):
rm = RegisteredModel("model_11")
mv = ModelVersion(rm, 8)
self.store.get_model_version_download_uri(model_version=mv)
self._verify_requests(
mock_http, "model-versions/get-download-uri", "POST",
GetModelVersionDownloadUri(model_version=mv.to_proto()))
def test_get_model_version_download_uri(mock_get_request_message, mock_model_registry_store):
name = "model1"
version = "32"
mock_get_request_message.return_value = GetModelVersionDownloadUri(name=name, version=version)
mock_model_registry_store.get_model_version_download_uri.return_value = "some/download/path"
resp = _get_model_version_download_uri()
_, args = mock_model_registry_store.get_model_version_download_uri.call_args
assert args == {"name": name, "version": version}
assert json.loads(resp.get_data()) == {"artifact_uri": "some/download/path"}
def test_get_model_version_download_uri(mock_get_request_message, mock_model_registry_store):
rm = RegisteredModel("model1")
mv = ModelVersion(registered_model=rm, version=32)
mock_get_request_message.return_value = GetModelVersionDownloadUri(model_version=mv.to_proto())
mock_model_registry_store.get_model_version_download_uri.return_value = "some/download/path"
resp = _get_model_version_download_uri()
args, _ = mock_model_registry_store.get_model_version_download_uri.call_args
assert args == (mv, )
assert json.loads(resp.get_data()) == {"artifact_uri": "some/download/path"}
def get_model_version_download_uri(self, model_version):
"""
Get the download location in Model Registry for this model version.
:param model_version: :py:class:`mlflow.entities.model_registry.ModelVersion` object.
:return: A single URI location that allows reads for downloading.
"""
req_body = message_to_json(GetModelVersionDownloadUri(
model_version=model_version.to_proto()))
response_proto = self._call_endpoint(GetModelVersionDownloadUri, req_body)
return response_proto.artifact_uri
def get_model_version_download_uri(self, name, version):
"""
Get the download location in Model Registry for this model version.
NOTE: For first version of Model Registry, since the models are not copied over to another
location, download URI points to input source path.
:param name: Registered model name.
:param version: Registered model version.
:return: A single URI location that allows reads for downloading.
"""
req_body = message_to_json(GetModelVersionDownloadUri(name=name, version=str(version)))
response_proto = self._call_endpoint(GetModelVersionDownloadUri, req_body)
return response_proto.artifact_uri
def get_model_version_download_uri(self, name, version):
"""
Get the download location in Model Registry for this model version.
:param name: Registered model name.
:param version: Registered model version.
:return: A single URI location that allows reads for downloading.
"""
req_body = message_to_json(
GetModelVersionDownloadUri(name=name, version=str(version)))
response_proto = self._call_endpoint(GetModelVersionDownloadUri,
req_body)
return response_proto.artifact_uri
