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))
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))