From 7c79c0b140c88a529962945ec5b482fe90c55581 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Thu, 11 Jul 2024 23:58:47 +0200 Subject: [PATCH] Update. --- grids.py | 5 ++++- main.py | 2 ++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/grids.py b/grids.py index 7aec62c..cfc7d16 100755 --- a/grids.py +++ b/grids.py @@ -1044,7 +1044,10 @@ if __name__ == "__main__": # nb, nrow = 8, 2 # for t in grids.all_tasks(): - for t in [grids.task_puzzle]: + for t in [ + grids.task_replace_color, + grids.task_frame, + ]: print(t.__name__) prompts, answers = grids.generate_prompts_and_answers_(nb, tasks=[t]) grids.save_quizzes("/tmp", t.__name__, prompts[:nb], answers[:nb], nrow=nrow) diff --git a/main.py b/main.py index 8d18119..b88cbc4 100755 --- a/main.py +++ b/main.py @@ -255,6 +255,8 @@ elif args.problem == "grids": else: raise ValueError +problem.save_some_examples(args.result_dir) + quiz_machine = quiz_machine.QuizMachine( problem=problem, nb_train_samples=args.nb_train_samples, -- 2.39.5