From c28688cb9691d63ec3936894df1ef1823a1b8817 Mon Sep 17 00:00:00 2001 From: joesharratt1229 Date: Sun, 16 Feb 2025 09:07:56 +0000 Subject: [PATCH] reformatted basic airth question template --- reasoning_gym/arithmetic/basic_arithmetic.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/arithmetic/basic_arithmetic.py b/reasoning_gym/arithmetic/basic_arithmetic.py index 156314d9..bab95b63 100644 --- a/reasoning_gym/arithmetic/basic_arithmetic.py +++ b/reasoning_gym/arithmetic/basic_arithmetic.py @@ -224,11 +224,13 @@ class BasicArithmeticDataset(ProceduralDataset): def _format_question(self, rng: Random, expression: str) -> str: """Format the expression according to config style""" + base_question = "Return only the answer to the following question: {question} =" if self.config.format_style == "simple": - return f"{expression} =" + return base_question.format(question=expression) else: templates = ["What is {0}?", "Calculate {0}", "Solve {0}", "Evaluate the expression: {0}"] - return rng.choice(templates).format(expression) + question = rng.choice(templates).format(expression) + return base_question.format(question=question) # Register the dataset