Fix

langchain-ai · Oct 15, 2024 · 515c4ff · 515c4ff
1 parent eefe057
commit 515c4ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langgraph/langgraph/pregel/executor.py b/libs/langgraph/langgraph/pregel/executor.py
@@ -93,8 +93,8 @@ def __exit__(
             if cancel:
                 task.cancel()
         # wait for all tasks to finish
-        if tasks := {t for t in tasks if not t.done()}:
-            concurrent.futures.wait(tasks)
+        if pending := {t for t in tasks if not t.done()}:
+            concurrent.futures.wait(pending)
         # shutdown the executor
         self.stack.__exit__(exc_type, exc_value, traceback)
         # re-raise the first exception that occurred in a task