From 2a769fb39e12657a18aa51ca824eb1b92f566238 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 17 Sep 2024 17:51:18 -0700 Subject: [PATCH] Less fun --- libs/langgraph/langgraph/pregel/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 1cf5ec982..671b6f843 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -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: