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(curriculum): Make boundaries in curriculum more sensible (#407)

Browse source 
* init

* fix tests

* unify codeio

* filtered for libraries not present in reasoning-gym

* fix more bounds

* puzzle24

* knight swap curriculum

* fix number sorting

* fix attributes

* add validation of config in creation of dataset

* dry run for instantiating and validating the datasets

* remove unused imports

* fix curriculum tests to reference newly updated attribute names
This commit is contained in:
Zafir Stojanovski 2025-04-04 20:24:14 +02:00 • committed by GitHub
parent 7853263650
commit dced3bfc45
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
132 changed files with 1226 additions and 347 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_number_sequences.py
View file

@ -94,17 +94,17 @@ def test_number_sequence_curriculum():
base_cfg: NumberSequenceConfig = curriculum.generate_configuration(base_value)
assert base_cfg.seed == 42
assert base_cfg.size == 150
assert base_cfg.max_complexity == 1 # Default level (0) corresponds to complexity 1
assert base_cfg.max_complexity == 2
# Test attribute level increment
curriculum.increment_attr_level("max_complexity")
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.max_complexity == 2 # Level 1 corresponds to complexity 2
assert increased_cfg.max_complexity == 3
# Test attribute level increment again
curriculum.increment_attr_level("max_complexity")
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.max_complexity == 3 # Level 2 corresponds to complexity 3
assert increased_cfg.max_complexity == 4
# Test that other parameters remain unchanged
assert increased_cfg.seed == 42
@ -113,4 +113,4 @@ def test_number_sequence_curriculum():
# Test attribute level decrement
curriculum.decrement_attr_level("max_complexity")
decreased_cfg = curriculum.generate_configuration(base_value)
assert decreased_cfg.max_complexity == 2 # Back to level 1, complexity 2
assert decreased_cfg.max_complexity == 3