X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=world.py;h=118a470b29b159d97826dc895362db8e8d673ded;hb=61c98647a2d708c8f2c5f0d25bcf05df92e1233f;hp=89833e6582349727a9372f3ba15ea66eeb95c3b5;hpb=e8e9b3941f150b20aa9585f7fa0a1f5e2fe6f547;p=culture.git

diff --git a/world.py b/world.py
index 89833e6..118a470 100755
--- a/world.py
+++ b/world.py
@@ -41,7 +41,7 @@ def generate(
     f_end = torch.zeros(nb, height, width, dtype=torch.int64)
     n = torch.arange(f_start.size(0))
 
-    for n in range(nb):
+    for n in tqdm.tqdm(range(nb), dynamic_ncols=True, desc="world generation"):
         nb_fish = torch.randint(max_nb_obj, (1,)).item() + 1
         for c in torch.randperm(colors.size(0) - 2)[:nb_fish].sort().values:
             i, j = (