This commit is contained in:
Nuno Campos
2024-09-17 17:51:18 -07:00
parent 3a2a4e44b2
commit 2a769fb39e
+2 -1
View File
@@ -1245,7 +1245,8 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]):
def get_waiter() -> asyncio.Task[None]:
nonlocal waiter
if waiter is None or waiter.done():
return (waiter := loop.submit(stream.wait))
waiter = loop.submit(stream.wait)
return waiter
else:
return waiter
else: