From 0be6757c554ab40b08b4acfd90787a86f4c4cc5b Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Wed, 3 Jul 2024 20:08:44 +0300
Subject: [PATCH] Update.

---
 reasoning.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/reasoning.py b/reasoning.py
index 92699e8..003806a 100755
--- a/reasoning.py
+++ b/reasoning.py
@@ -307,7 +307,13 @@ class Reasoning(problem.Problem):
         prompts = torch.zeros(nb, self.height, self.width * 3, dtype=torch.int64)
         answers = torch.zeros(nb, self.height, self.width, dtype=torch.int64)
         w = self.width
-        for prompt, answer in zip(prompts, answers):
+
+        for prompt, answer in tqdm.tqdm(
+            zip(prompts, answers),
+            dynamic_ncols=True,
+            desc="world generation",
+            total=prompts.size(0),
+        ):
             A = prompt[:, 0 * w : 1 * w]
             f_A = prompt[:, 1 * w : 2 * w]
             B = prompt[:, 2 * w : 3 * w]
-- 
2.39.5