Skip to content

Commit 3d1dc5b

Browse files
authored
Fix generator wrapper of concurrency guard (#995)
1 parent 484bfdc commit 3d1dc5b

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

src/neo4j/_async/_debug/_concurrency_check.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,8 @@ async def inner(*args, **kwargs):
131131
if acquired:
132132
try:
133133
item = await iter_.__anext__()
134+
except StopAsyncIteration:
135+
return
134136
finally:
135137
async with self.__tracebacks_lock:
136138
self.__tracebacks.pop()

src/neo4j/_sync/_debug/_concurrency_check.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,8 @@ def inner(*args, **kwargs):
131131
if acquired:
132132
try:
133133
item = iter_.__next__()
134+
except StopIteration:
135+
return
134136
finally:
135137
with self.__tracebacks_lock:
136138
self.__tracebacks.pop()

0 commit comments

Comments
 (0)