From 6dbb5b39ee950842296fb716d9f2825d44d2e5cb Mon Sep 17 00:00:00 2001 From: "Andreas Koepf (aider)" Date: Thu, 23 Jan 2025 14:22:46 +0100 Subject: [PATCH] refactor: Reorder parameters of _format_question to match other methods --- reasoning_gym/arithmetic/basic_arithmetic.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/arithmetic/basic_arithmetic.py b/reasoning_gym/arithmetic/basic_arithmetic.py index 19cd947c..7ea7027a 100644 --- a/reasoning_gym/arithmetic/basic_arithmetic.py +++ b/reasoning_gym/arithmetic/basic_arithmetic.py @@ -64,7 +64,7 @@ class ArithmeticDataset: else: expression, result = self._generate_simple_task(item_rng, num_terms, num_digits) - question = self._format_question(expression, item_rng) + question = self._format_question(item_rng, expression) return { "question": question, @@ -156,7 +156,7 @@ class ArithmeticDataset: self._current_idx += 1 return item - def _format_question(self, expression: str, rng: Random) -> str: + def _format_question(self, rng: Random, expression: str) -> str: """Format the expression according to config style""" if self.config.format_style == "simple": return f"{expression} ="