Return resource object in get_batches.
Co-authored-by: p-pautov <37922380+p-pautov@users.noreply.github.com>
This commit is contained in:
parent
41bf12e56e
commit
e402206a75
1 changed files with 2 additions and 1 deletions
|
|
@ -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()
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue