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)