Skip to content

Commit d4b36ff

Browse files
committed
Ensure rich progress bar has task ID assigned before updating
1 parent 583552e commit d4b36ff

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sqlmesh/core/console.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1060,6 +1060,7 @@ def update_snapshot_evaluation_progress(
10601060
"""Update the snapshot evaluation progress."""
10611061
if (
10621062
self.evaluation_total_progress
1063+
and self.evaluation_total_task is not None
10631064
and self.evaluation_model_progress
10641065
and self.evaluation_progress_live
10651066
):
@@ -1102,7 +1103,7 @@ def update_snapshot_evaluation_progress(
11021103
self.evaluation_progress_live.console.print(msg)
11031104

11041105
self.evaluation_total_progress.update(
1105-
self.evaluation_total_task or TaskID(0), refresh=True, advance=1
1106+
self.evaluation_total_task, refresh=True, advance=1
11061107
)
11071108

11081109
model_task_id = self.evaluation_model_tasks[snapshot.name]

0 commit comments

Comments
 (0)