aboutsummaryrefslogtreecommitdiff
path: root/scripts/prompt_matrix.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/prompt_matrix.py')
-rw-r--r--scripts/prompt_matrix.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/scripts/prompt_matrix.py b/scripts/prompt_matrix.py
index ed31dd36..720d7583 100644
--- a/scripts/prompt_matrix.py
+++ b/scripts/prompt_matrix.py
@@ -20,6 +20,8 @@ def draw_xy_grid(xs, ys, x_label, y_label, cell):
first_pocessed = None
+ state.job_count = len(xs) * len(ys)
+
for iy, y in enumerate(ys):
for ix, x in enumerate(xs):
state.job = f"{ix + iy * len(xs) + 1} out of {len(xs) * len(ys)}"
@@ -29,6 +31,7 @@ def draw_xy_grid(xs, ys, x_label, y_label, cell):
first_pocessed = processed
res.append(processed.images[0])
+ state.nextjob()
grid = images.image_grid(res, rows=len(ys))
grid = images.draw_grid_annotations(grid, res[0].width, res[0].height, hor_texts, ver_texts)