def CreateModel(self, request, context): """Creates a new model from an import source.""" model = self.explainer.CreateModel(request.type, request.name) reply = explain_pb2.CreateModelReply( model=explain_pb2.Model(name=model.name, handle=model.handle, status=model.state, message=model.message)) return reply
def ListModel(self, request, _): """List all models.""" models = self.explainer.ListModel() models_pb = [] for model in models: models_pb.append( explain_pb2.Model(name=model.name, handle=model.handle, status=model.state, message=model.message)) reply = explain_pb2.ListModelReply() reply.models.extend(models_pb) return reply