From e402206a75f127798be263fafac7d57da644b20b Mon Sep 17 00:00:00 2001 From: Eugene <54681898+jimf5@users.noreply.github.com> Date: Thu, 19 Dec 2024 17:04:36 -0800 Subject: [PATCH] Return resource object in get_batches. Co-authored-by: p-pautov <37922380+p-pautov@users.noreply.github.com> --- tests/trace_service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/trace_service.py b/tests/trace_service.py index 47e477d..3b191a1 100644 --- a/tests/trace_service.py +++ b/tests/trace_service.py @@ -20,7 +20,8 @@ class TraceService(trace_service_pb2_grpc.TraceServiceServicer): break time.sleep(0.001) assert len(self.batches) == 1 - return self.batches.pop() + assert len(self.batches[0]) == 1 + return self.batches.pop()[0] def get_span(self): batch = self.get_batch()