open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-28 17:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Ensure consistent variable usage in simple equations generation

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 15:38:26 +01:00
parent aa55f1d6f2
commit 2662a2410b
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
reasoning_gym/algebra/simple_equations.py
View file

@ -48,15 +48,16 @@ class SimpleEquationsDataset(ProceduralDataset):
"""
rng = random.Random(self.seed + idx)
# Generate equation and solution
equation, solution = self._generate_equation(rng)
# Get variable and generate equation
variable = self._get_variable(rng)
equation, solution = self._generate_equation(rng, variable)
return {
"question": equation,
"answer": str(solution),
"metadata": {
"equation": equation,
"variable": self._get_variable(rng),
"variable": variable,
}
}
@ -64,14 +65,17 @@ class SimpleEquationsDataset(ProceduralDataset):
"""Get a random lowercase variable name"""
return rng.choice(string.ascii_lowercase)
def _generate_equation(self, rng: random.Random) -> Tuple[str, int]:
def _generate_equation(self, rng: random.Random, variable: str) -> Tuple[str, int]:
"""Generate an equation and its solution
Args:
rng: Random number generator
variable: Variable symbol to use in equation
Returns:
Tuple of (equation string, solution integer)
"""
var = self._get_variable(rng)
x = Symbol(var)
x = Symbol(variable)
# Generate left side
num_terms = rng.randint(self.config.min_terms, self.config.max_terms)