This commit is contained in:
Nuno Campos
2025-06-25 17:07:26 -07:00
parent 3ddb6b1477
commit cee9ac0b7a
+2 -6
View File
@@ -1587,9 +1587,7 @@ class Pregel(PregelProtocol[StateT, InputT, OutputT], Generic[StateT, InputT, Ou
step + 4,
for_execution=True,
store=self.store,
checkpointer=self.checkpointer
if isinstance(self.checkpointer, BaseCheckpointSaver)
else None,
checkpointer=checkpointer,
manager=None,
)
@@ -2045,9 +2043,7 @@ class Pregel(PregelProtocol[StateT, InputT, OutputT], Generic[StateT, InputT, Ou
step + 4,
for_execution=True,
store=self.store,
checkpointer=self.checkpointer
if isinstance(self.checkpointer, BaseCheckpointSaver)
else None,
checkpointer=checkpointer,
manager=None,
)