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

Add attributes for curriculum

Browse source 
Co-authored-by: EduardDurech <39579228+EduardDurech@users.noreply.github.com>
This commit is contained in:
Andreas Koepf 2025-02-10 18:58:07 +01:00
parent bea9e6d96a
commit 8772041afb
6 changed files with 246 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_chain_sum.py
View file

@ -1,6 +1,7 @@
import pytest
from reasoning_gym.arithmetic import ChainSum, ChainSumConfig
from reasoning_gym.arithmetic.chain_sum import ChainSumCurriculum
def test_chain_sum_config_validation():
@ -127,3 +128,20 @@ def test_chain_sum_iteration():
first_items = list(dataset)
second_items = list(dataset)
assert first_items == second_items, "Multiple iterations should yield same items"
def test_chain_sum_curriculum():
c = ChainSumCurriculum()
base_value = {"size": 150, "seed": 1}
base_cfg: ChainSumConfig = c.generate_configuration(base_value)
assert base_cfg.seed == 1
assert base_cfg.size == 150
assert base_cfg.min_digits == 1 and base_cfg.max_digits == 1
assert base_cfg.min_terms == 2 and base_cfg.max_terms == 2
c.increment_attr_level("num_terms")
c.increment_attr_level("num_digits")
config2 = c.generate_configuration()