Update.

[culture.git] / grids.py

diff --git a/grids.py b/grids.py
index 659bd6c..ed72099 100755 (executable)
--- a/grids.py
+++ b/grids.py
@@ -647,7 +647,7 @@ class Grids(problem.Problem):
         S = self.height * self.width
         Bs = prompts[:, 2 * (S + 1) : 2 * (S + 1) + S]
         f_Bs = answers
-        return (B_s == f_Bs).long().min(dim=-1).values > 0
+        return (Bs == f_Bs).long().min(dim=-1).values > 0
 
     def generate_prompts_and_answers(self, nb, tasks=None, device="cpu"):
         if tasks is None: