lib: Unset skip_done_tasks after each tick of the loop

This commit is contained in:
Nuno Campos
2024-11-04 11:23:53 -08:00
parent 58d7eb9b17
commit 014f8485a6
+3
View File
@@ -399,6 +399,9 @@ class PregelLoop(LoopProtocol):
for task in self.tasks.values():
if task.writes:
self._output_writes(task.id, task.writes, cached=True)
elif not self.skip_done_tasks:
# "not skip_done_tasks" only applies to first tick after resuming
self.skip_done_tasks = True
# if all tasks have finished, re-tick
if all(task.writes for task in self.tasks.values()):