Update.

diff --git a/grids.py b/grids.py
index 7aec62c..cfc7d16 100755 (executable)
--- 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 (executable)
--- 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,