open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: Reorder parameters of _format_question to match other methods

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 14:22:46 +01:00
parent f12e3788d1
commit 6dbb5b39ee
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -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} ="