X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=38dccb9f8eea57437ce7574d4f87208ab0077b38;hb=7d596f5ca37e7e9ac53047acdc7d00928821929f;hp=bb1e7b46524f0d170e233e33c3335965e5588636;hpb=4502a109727b0424ff6d4df90f17b361524f9e73;p=picoclvr.git

diff --git a/main.py b/main.py
index bb1e7b4..38dccb9 100755
--- a/main.py
+++ b/main.py
@@ -968,7 +968,9 @@ class TaskStack(Task):
             )
 
             #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
-            input = self.test_input[:10, :50]
+            l = 50
+            l = l - l % (1 + self.nb_digits)
+            input = self.test_input[:10, :l]
             result = input.clone()
             stack.remove_popped_values(result, self.nb_stacks, self.nb_digits)
             ar_mask = (result != input).long()