def GetResults(self, request, context):
if request.comp_id >= len(self.results):
return hetr_pb2.GetResultsReply(status=False)
pb_results = []
for r in self.results[request.comp_id]:
pb_val = hetr_pb2.Value()
if is_scalar_type(r):
assign_scalar(pb_val.scalar, r)
else:
pb_val.tensor.CopyFrom(tensor_to_protobuf(r))
pb_results.append(pb_val)
return hetr_pb2.GetResultsReply(status=True, results=pb_results)
def GetResults(self, request, context):
logger.debug("server: get_results")
if request.comp_id not in self.results:
message = 'unknown computation id {}'.format(request.comp_id)
return hetr_pb2.GetResultsReply(status=False, message=message)
try:
pb_results = []
for r in self.results[request.comp_id]:
pb_val = hetr_pb2.Value()
if is_scalar_type(r):
assign_scalar(pb_val.scalar, r)
else:
pb_val.tensor.CopyFrom(tensor_to_protobuf(r))
pb_results.append(pb_val)
return hetr_pb2.GetResultsReply(status=True, results=pb_results)
except Exception:
return hetr_pb2.GetResultsReply(status=False, message=traceback.format_exc())