This commit is contained in:
Nuno Campos
2025-01-15 15:29:34 -08:00
parent 5375af7827
commit 71fbd6a8b4
+6 -2
View File
@@ -494,7 +494,9 @@ class Pregel(PregelProtocol):
saved.metadata.get("step", -1) + 1,
for_execution=True,
store=self.store,
checkpointer=self.checkpointer or None,
checkpointer=self.checkpointer
if isinstance(self.checkpointer, BaseCheckpointSaver)
else None,
manager=None,
)
# get the subgraphs
@@ -606,7 +608,9 @@ class Pregel(PregelProtocol):
saved.metadata.get("step", -1) + 1,
for_execution=True,
store=self.store,
checkpointer=self.checkpointer or None,
checkpointer=self.checkpointer
if isinstance(self.checkpointer, BaseCheckpointSaver)
else None,
manager=None,
)
# get the subgraphs