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 #484] resolve basic_arithmetic fails when size is large (#485)

Browse source 
* [fix] resolve basic_arithmetic fails when size is large by replacing zero divisor with 1
This commit is contained in:
theblackcat102 2025-07-07 16:46:23 +08:00 • committed by GitHub
parent bf451d5197
commit 2d19f13e0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_basic_arithmetic.py
View file

@ -168,3 +168,19 @@ def test_basic_arithmetic_curriculum_upper_bound():
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.min_terms == 2 and increased_cfg.max_terms == 3
assert increased_cfg.min_digits == 1 and increased_cfg.max_digits == 2
def test_arithmetic_dataset_large_random_generation():
"""Stress-test generation of many arithmetic questions to catch random errors"""
config = BasicArithmeticDatasetConfig(
size=100000,
seed=123,
min_terms=2,
max_terms=6,
min_digits=1,
max_digits=3,
)
dataset = BasicArithmeticDataset(config)
for item in dataset:
assert isinstance(item, dict)
assert "question" in item and "answer" in item and "metadata" in item