Retry condition for resuming flag should apply only to top graphs

2026-08-22 07:32:25 +02:00 · 2025-03-04 09:51:13 -08:00
parent 48164a95da
commit 137dcce5b5
1 changed files with 5 additions and 2 deletions
@@ -569,8 +569,11 @@ class PregelLoop(LoopProtocol):
                CONFIG_KEY_RESUMING,
                self.input is None
                or isinstance(self.input, Command)
-                or self.config.get("metadata", {}).get("run_id")
-                == self.checkpoint_metadata.get("run_id", MISSING),
+                or (
+                    not self.is_nested
+                    and self.config.get("metadata", {}).get("run_id")
+                    == self.checkpoint_metadata.get("run_id", MISSING)
+                ),
            )
        )