In update state skip nodes not found in instance

eg, because removed since
2026-08-24 08:32:24 +02:00 · 2024-09-10 16:17:22 -07:00
parent 5e6b58f847
commit 1ce8d628f2
1 changed files with 2 additions and 0 deletions
@@ -782,6 +782,7 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]):
            last_seen_by_node = sorted(
                (v, n)
                for n, seen in checkpoint["versions_seen"].items()
+                if n in self.nodes
                for v in seen.values()
            )
            # if two nodes updated the state at the same time, it's ambiguous
@@ -939,6 +940,7 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]):
            last_seen_by_node = sorted(
                (v, n)
                for n, seen in checkpoint["versions_seen"].items()
+                if n in self.nodes
                for v in seen.values()
            )
            # if two nodes updated the state at the same time, it's ambiguous