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