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

fix: Prevent infinite loop in arithmetic dataset space generation

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 11:34:18 +01:00
parent 473a2be31d
commit 6b41e3cc47
2 changed files with 9 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

@ -14,7 +14,7 @@ class ArithmeticDatasetConfig:
allow_parentheses: bool = True
allow_negation: bool = True
seed: Optional[int] = None
size: int = 10000 # Virtual dataset size
size: int = 500 # Virtual dataset size
format_style: Literal["simple", "natural"] = "simple"
def validate(self):
@ -102,14 +102,14 @@ class ArithmeticDataset:
add_terms(num_terms)
# Add random spaces
# Add at most one random space between parts
space_parts = []
for p in parts:
while rng.random() < 0.15:
if rng.random() < 0.15:
space_parts.append(" ")
space_parts.append(p)
expression = " ".join(space_parts)
expression = " ".join(space_parts).strip()
result = eval(expression) # Note: eval is safe here as we control the input
return expression, result