This commit is contained in:
Nuno Campos
2024-09-19 11:46:21 -07:00
parent cc318b1156
commit 4f767cd2ca
+1 -1
View File
@@ -1429,7 +1429,7 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]):
# set up messages stream mode
if "messages" in stream_modes:
run_manager.inheritable_handlers.append(
StreamMessagesHandler(stream.put)
StreamMessagesHandler(stream.put_nowait)
)
async with AsyncPregelLoop(
input,