X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=tasks.py;h=cbc8e6b05c81aab7f6c696b8304574756619ac38;hb=3e3b9ead54130e5e3b2ce690943af9cb4c894e65;hp=24c13fe79eddb4b8104bfe869c079bb4cbce80b9;hpb=5332c56acd44d7049f3fbb33a8643482e0c71f4d;p=picoclvr.git

diff --git a/tasks.py b/tasks.py
index 24c13fe..cbc8e6b 100755
--- a/tasks.py
+++ b/tasks.py
@@ -1495,8 +1495,8 @@ class Grid(Task):
         self.token2id = dict([(t, n) for n, t in enumerate(tokens)])
         self.id2token = dict([(n, t) for n, t in enumerate(tokens)])
         self.t_nul = self.token2id["#"]
-        self.t_true = self.token2id["<true>"]
-        self.t_false = self.token2id["<false>"]
+        self.t_true = self.token2id["true"]
+        self.t_false = self.token2id["false"]
 
         # Tokenize the train and test sets
         self.train_input = self.tensorize(self.train_descr)