From 525bd24014786b53638dea78cfb88035a2b99d97 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 29 Jun 2024 16:24:46 +0300 Subject: [PATCH] Update. --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index b88847e..5537965 100755 --- a/main.py +++ b/main.py @@ -221,10 +221,10 @@ else: assert args.nb_train_samples % args.batch_size == 0 assert args.nb_test_samples % args.batch_size == 0 -if args.problem=="sky": - problem=sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2, speed=2), -elif args.problem="wireworld": - problem=wireworld.Wireworld(height=10, width=15, nb_iterations=4) +if args.problem == "sky": + problem = (sky.Sky(height=6, width=8, nb_birds=3, nb_iterations=2, speed=2),) +elif args.problem == "wireworld": + problem = wireworld.Wireworld(height=10, width=15, nb_iterations=4) else: raise ValueError -- 2.20.1