fix progress when completed (#1239)

eakmanrq · web-flow · commit 545baae61722 · 2023-07-31T13:17:37.000-07:00
diff --git a/sqlmesh/core/console.py b/sqlmesh/core/console.py
@@ -1190,7 +1190,7 @@ def update_snapshot_evaluation_progress(self, snapshot: Snapshot, num_batches: i
                 [
                     s
                     for s, total in self.evaluation_batches.items()
-                    if self.evaluation_batch_progress[s.name][1] == total
+                    if self.evaluation_batch_progress.get(s.name, (None, -1))[1] == total
                 ]
             )
             total = len(self.evaluation_batch_progress)

Original file line number	Diff line number	Diff line change
`@@ -1190,7 +1190,7 @@ def update_snapshot_evaluation_progress(self, snapshot: Snapshot, num_batches: i`
`1190`	`1190`	`[`
`1191`	`1191`	`s`
`1192`	`1192`	`for s, total in self.evaluation_batches.items()`
`1193`		`- if self.evaluation_batch_progress[s.name][1] == total`
	`1193`	`+ if self.evaluation_batch_progress.get(s.name, (None, -1))[1] == total`
`1194`	`1194`	`]`
`1195`	`1195`	`)`
`1196`	`1196`	`total = len(self.evaluation_batch_progress)`