def _update_model_version():
request_message = _get_request_message(UpdateModelVersion())
new_description = None
if request_message.HasField("description"):
new_description = request_message.description
model_version = _get_model_registry_store().update_model_version(
name=request_message.name, version=request_message.version, description=new_description)
return _wrap_response(UpdateModelVersion.Response(model_version=model_version.to_proto()))
def _update_model_version():
request_message = _get_request_message(UpdateModelVersion())
new_stage = None
new_description = None
if request_message.HasField("stage"):
new_stage = request_message.stage
if request_message.HasField("description"):
new_description = request_message.description
_get_model_registry_store().update_model_version(
ModelVersion.from_proto(request_message.model_version), new_stage,
new_description)
return _wrap_response(UpdateModelVersion.Response())
def _update_model_version():
request_message = _get_request_message(UpdateModelVersion())
new_description = None
if request_message.HasField("description"):
new_description = request_message.description
model_version = _get_model_registry_store().update_model_version(
name=request_message.name,
version=request_message.version,
description=new_description)
if request_message.HasField("stage"):
model_version = _get_model_registry_store(
).transition_model_version_stage(name=request_message.name,
version=request_message.version,
stage=request_message.stage,
archive_existing_versions=False)
return _wrap_response(
UpdateModelVersion.Response(model_version=model_version.to_proto()))
