Retry condition for resuming flag should apply only to top graphs (#3676)

2026-09-05 17:27:47 +02:00 · 2025-03-04 10:20:35 -08:00
parent 48164a95da 137dcce5b5
commit 03bf149ebd
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)
+                ),
            )
        )