def _update_registered_model():
request_message = _get_request_message(UpdateRegisteredModel())
name = request_message.name
new_description = request_message.description
registered_model = _get_model_registry_store().update_registered_model(
name=name, description=new_description)
response_message = UpdateRegisteredModel.Response(registered_model=registered_model.to_proto())
return _wrap_response(response_message)
def _update_registered_model():
request_message = _get_request_message(UpdateRegisteredModel())
new_name = None
new_description = None
if request_message.HasField("name"):
new_name = request_message.name
if request_message.HasField("description"):
new_description = request_message.description
registered_model = _get_model_registry_store().update_registered_model(
RegisteredModel.from_proto(request_message.registered_model), new_name,
new_description)
response_message = UpdateRegisteredModel.Response(
registered_model=registered_model.to_proto())
return _wrap_response(response_message)