X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=ff573c4ebff59f8ca8ee42a814a098cd0a58999e;hb=3ea12df1dcfc4006eb895fd62bb622e9aef6178c;hp=02e1a8d6698a58d84c92001b8a06bfba2a461d47;hpb=3e1100f546e955430d87dd6808c8d148715bc50d;p=culture.git

diff --git a/main.py b/main.py
index 02e1a8d..ff573c4 100755
--- a/main.py
+++ b/main.py
@@ -13,7 +13,7 @@ from torch.nn import functional as F
 
 import ffutils
 import mygpt
-import sky, reasoning, quiz_machine
+import sky, grids, quiz_machine
 
 # world quizzes vs. culture quizzes
 
@@ -251,8 +251,8 @@ if args.problem == "sky":
         speed=args.sky_speed,
     )
     back_accuracy = False
-elif args.problem == "reasoning":
-    problem = reasoning.Reasoning(device=device)
+elif args.problem == "grids":
+    problem = grids.Grids(device=device)
     back_accuracy = True
 else:
     raise ValueError