Skip to content

Commit a3c6f3b

Browse files
committed
Fix scheduler rebuild during task generation
1 parent 55e9ad8 commit a3c6f3b

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

src/_pytask/execute.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,9 @@ def pytask_execute_log_start(session: Session) -> None:
8686
@hookimpl
8787
def pytask_execute_build(session: Session) -> bool | None:
8888
"""Execute tasks."""
89-
scheduler = session.scheduler
90-
if scheduler is not None:
91-
while scheduler.is_active():
92-
task_name = scheduler.get_ready()[0]
89+
if session.scheduler is not None:
90+
while session.scheduler.is_active():
91+
task_name = session.scheduler.get_ready()[0]
9392
task = session.dag.nodes[task_name]
9493
if not isinstance(task, PTask):
9594
msg = f"Expected task node for signature {task_name!r}."
@@ -98,7 +97,7 @@ def pytask_execute_build(session: Session) -> bool | None:
9897
session=session, task=task
9998
)
10099
session.execution_reports.append(report)
101-
scheduler.done(task_name)
100+
session.scheduler.done(task_name)
102101

103102
if session.should_stop:
104103
return True

0 commit comments

Comments
 (0)