X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=7f9d5210a486971126dcbf948224d8f296f79bf1;hb=7ad1043be4c7f85625e164fd586bc71096f93e5b;hp=2d5b14891aa00487794868054a2630a6fead6e33;hpb=9f787901b2c7591a323f843ab973fe6abcf6b8ce;p=culture.git

diff --git a/main.py b/main.py
index 2d5b148..7f9d521 100755
--- 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,