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