From 07c065e77f1d2a775814ec402752a4a8eb6c7574 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Thu, 11 Jul 2024 08:40:39 +0200
Subject: [PATCH] Update.
---
grids.py | 9 +++++++++
main.py | 2 ++
problem.py | 3 +++
3 files changed, 14 insertions(+)
diff --git a/grids.py b/grids.py
index d1653ee..20a964b 100755
--- a/grids.py
+++ b/grids.py
@@ -942,6 +942,15 @@ class Grids(problem.Problem):
nrow,
)
+ def save_some_examples(self, result_dir):
+ nb, nrow = 72, 4
+ for t in self.all_tasks():
+ print(t.__name__)
+ prompts, answers = self.generate_prompts_and_answers_(nb, tasks=[t])
+ self.save_quizzes(
+ result_dir, t.__name__, prompts[:nb], answers[:nb], nrow=nrow
+ )
+
######################################################################
diff --git a/main.py b/main.py
index e6806d4..5956be5 100755
--- a/main.py
+++ b/main.py
@@ -253,6 +253,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,
diff --git a/problem.py b/problem.py
index 617b2a8..7eeb6b4 100755
--- a/problem.py
+++ b/problem.py
@@ -88,3 +88,6 @@ class Problem:
prompts, answers = prompts[:-k], answers[:-k]
return prompts, answers
+
+ def save_some_examples(self, result_dir):
+ pass
--
2.39.5