From b0f0d6f3f7465dafc8da98be5c9bd86d8dd1fd21 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Wed, 7 Feb 2024 19:49:35 -0800
Subject: [PATCH] Fix errors being swallowed

---
 langgraph/pregel/__init__.py | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py
index ad0b650d2..016f3a35b 100644
--- a/langgraph/pregel/__init__.py
+++ b/langgraph/pregel/__init__.py
@@ -392,11 +392,10 @@ class Pregel(
         finally:
             # cancel any pending tasks when generator is interrupted
             try:
-                futures
+                for task in futures:
+                    task.cancel()
             except NameError:
-                return
-            for task in futures:
-                task.cancel()
+                pass
 
     async def _atransform(
         self,
@@ -562,11 +561,10 @@ class Pregel(
         finally:
             # cancel any pending tasks when generator is interrupted
             try:
-                futures
+                for task in futures:
+                    task.cancel()
             except NameError:
-                return
-            for task in futures:
-                task.cancel()
+                pass
 
     def invoke(
         self,