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 40596262e1
commit fbba398c91
2 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_arithmetic.py
View file

@ -58,7 +58,11 @@ def test_arithmetic_dataset_format_styles():
config = ArithmeticDatasetConfig(
size=10,
seed=42,
format_style="simple"
format_style="simple",
min_terms=2,
max_terms=3, # Keep expressions simple for testing
min_digits=1,
max_digits=2
)
dataset = ArithmeticDataset(config)
assert all(item["question"].endswith("=") for item in dataset)