Update.
authorFrançois Fleuret <francois@fleuret.org>
Tue, 25 Jun 2024 19:02:03 +0000 (21:02 +0200)
committerFrançois Fleuret <francois@fleuret.org>
Tue, 25 Jun 2024 19:02:03 +0000 (21:02 +0200)
main.py

diff --git a/main.py b/main.py
index 2d5b148..7f9d521 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -211,7 +211,7 @@ assert args.nb_train_samples % args.batch_size == 0
 assert args.nb_test_samples % args.batch_size == 0
 
 quizz_machine = quizz_machine.QuizzMachine(
-    sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2),
+    problem=sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2),
     nb_train_samples=args.nb_train_samples,
     nb_test_samples=args.nb_test_samples,
     batch_size=args.physical_batch_size,