def getModelVersionByVersion(self, request, context):
model_version_relation = self.store.get_model_version_relation_by_version(request.name, request.model_id)
if model_version_relation is None:
model_version_detail = None
else:
model_relation = self.store.get_model_relation_by_id(model_version_relation.model_id)
model_version_detail = self.model_center_client.get_model_version_detail(model_relation.name, request.name)
return _wrap_meta_response(
MetaToProto.model_version_meta_to_proto(model_version_relation, model_version_detail))
def registerModelVersion(self, request, context):
model_version = transform_model_version_meta(request.model_version)
model_relation = self.store.get_model_relation_by_id(model_version.model_id)
model_version_detail = self.model_center_client.create_model_version(model_relation.name,
model_version.model_path,
model_version.model_metric,
model_version.model_flavor,
model_version.version_desc,
request.model_version.current_stage)
model_version_relation = self.store.register_model_version_relation(version=model_version_detail.model_version,
model_id=model_version.model_id,
workflow_execution_id=
model_version.workflow_execution_id)
return _wrap_meta_response(
MetaToProto.model_version_meta_to_proto(model_version_relation, model_version_detail))
