Fix errors being swallowed

2026-08-28 18:59:42 +02:00 · 2024-02-07 19:49:35 -08:00
parent f2ad930cd4
commit b0f0d6f3f7
1 changed files with 6 additions and 8 deletions
@@ -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,