Update.

[culture.git] / main.py

diff --git a/main.py b/main.py
index 8b5d9a4..a954af6 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -250,7 +250,7 @@ if args.problem == "sky":
         speed=args.sky_speed,
     )
 elif args.problem == "reasoning":
-    problem = reasoning.Reasoning()
+    problem = reasoning.Reasoning(device=device)
 else:
     raise ValueError