示例#1
0
 def getModelById(self, request, context):
     model_relation = self.store.get_model_relation_by_id(request.id)
     if model_relation is None:
         model_detail = None
     else:
         model_detail = self.model_center_client.get_registered_model_detail(model_relation.name)
     return _wrap_meta_response(MetaToProto.model_meta_to_proto(model_relation, model_detail))
 def registerModel(self, request, context):
     model = transform_model_meta(request.model)
     model_detail = self.model_center_client.create_registered_model(
         model.name, ModelType.Name(model.model_type), model.model_desc)
     model_relation = self.store.register_model_relation(
         name=model.name, project_id=model.project_id)
     return _wrap_meta_response(
         MetaToProto.model_meta_to_proto(model_relation, model_detail))
示例#3
0
 def getModelByName(self, request, context):
     model_relation = self.store.get_model_relation_by_name(request.name)
     model_detail = self.model_center_client.get_registered_model_detail(request.name)
     return _wrap_meta_response(MetaToProto.model_meta_to_proto(model_relation, model_detail))