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: Pass item_rng to _format_question() method in ArithmeticDataset

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 14:20:22 +01:00
parent cdb343ac8d
commit dce4c7eba5
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
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)
question = self._format_question(expression, item_rng)
return {
"question": question,
@ -156,15 +156,13 @@ class ArithmeticDataset:
self._current_idx += 1
return item
def _format_question(self, expression: str) -> str:
def _format_question(self, expression: str, rng: Random) -> str:
"""Format the expression according to config style"""
if self.config.format_style == "simple":
return f"{expression} ="
else:
templates = ["What is {0}?", "Calculate {0}", "Solve {0}", "Evaluate the expression: {0}"]
# Use deterministic RNG for template selection
template_rng = Random(self.seed)
return template_rng.choice(templates).format(expression)
return rng.choice(templates).format(expression)
def arithmetic_dataset(