From e6025f0e35504e37fe25ceee9eed130043220fd5 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Sun, 21 Jul 2024 08:17:11 +0200
Subject: [PATCH] Update.

---
 grids.py        | 2 ++
 quiz_machine.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/grids.py b/grids.py
index 273f84e..0b789ef 100755
--- a/grids.py
+++ b/grids.py
@@ -221,6 +221,8 @@ class Grids(problem.Problem):
             self.task_scale,
             self.task_symbols,
             self.task_isometry,
+            self.task_corners,
+            self.task_proximity,
             # self.task_islands, # TOO MESSY
         ]
 
diff --git a/quiz_machine.py b/quiz_machine.py
index cbaa7cd..b965e33 100755
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -338,7 +338,7 @@ class QuizMachine:
 
         test_result, test_correct = compute_accuracy(input, log_prefix="test")
 
-        n_test_p2a = model.test_w_quizzes[:2000, 0] == self.problem.token_forward
+        n_test_p2a = input[:, 0] == self.problem.token_forward
 
         p2a_test_correct = test_correct[n_test_p2a]
 
-- 
2.39.5