Return resource object in get_batches.

Co-authored-by: p-pautov <37922380+p-pautov@users.noreply.github.com>
This commit is contained in:
Eugene 2024-12-19 17:04:36 -08:00 committed by GitHub
parent 41bf12e56e
commit e402206a75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -20,7 +20,8 @@ class TraceService(trace_service_pb2_grpc.TraceServiceServicer):
break break
time.sleep(0.001) time.sleep(0.001)
assert len(self.batches) == 1 assert len(self.batches) == 1
return self.batches.pop() assert len(self.batches[0]) == 1
return self.batches.pop()[0]
def get_span(self): def get_span(self):
batch = self.get_batch() batch = self.get_batch()