This commit is contained in:
Tat Dat Duong
2025-03-19 14:21:43 +01:00
parent e6726802f7
commit 1e751a2256
+2 -2
View File
@@ -1485,7 +1485,7 @@ class Pregel(PregelProtocol):
),
)
for _, task in tasks:
for task_id, task in tasks:
if push_writes := [w for w in task.writes if w[0] == PUSH]:
checkpointer.put_writes(next_config, push_writes, task_id)
@@ -1818,7 +1818,7 @@ class Pregel(PregelProtocol):
),
)
for _, task in tasks:
for task_id, task in tasks:
# save push writes
if push_writes := [w for w in task.writes if w[0] == PUSH]:
await checkpointer.aput_writes(next_config, push_writes, task_id)