open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor Curriculum Attributes (#335)

Browse source 
* remove min_value from AttributeDefinition
* remove type from AttributeDefinition
* Add CurriculumContext
* add ensure_interval option for RangeAttributes
* docs: Add legend explaining curriculum indicators in dataset gallery
* update GALLERY.md
This commit is contained in:
Andreas Köpf 2025-03-16 15:40:28 +01:00 • committed by GitHub
parent 4e7d9296ee
commit d2c895f1d3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
101 changed files with 286 additions and 677 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_graph_color.py
View file

@ -3,6 +3,7 @@ import json
import pytest
from reasoning_gym.algorithmic.graph_color import GraphColorConfig, GraphColorCurriculum, GraphColorDataset
from reasoning_gym.coaching.base_curriculum import DefaultCurriculumContext, RangeAttributeMode
def test_graph_color():
@ -79,14 +80,15 @@ def test_graph_color_curriculum():
base_value = {"size": 150, "seed": 1}
base_cfg: GraphColorConfig = curriculum.generate_configuration(base_value)
context = DefaultCurriculumContext(mode=RangeAttributeMode.UPPER_BOUND)
base_cfg: GraphColorConfig = curriculum.generate_configuration(base_value, context=context)
assert base_cfg.size == 150
assert base_cfg.seed == 1
assert base_cfg.min_num_vertices == base_cfg.max_num_vertices == 10
assert base_cfg.num_colors == base_cfg.num_colors == 5
curriculum.increment_attr_level("num_vertices")
cfg = curriculum.generate_configuration(base_value)
cfg = curriculum.generate_configuration(base_value, context=context)
assert cfg.min_num_vertices == 20
curriculum.increment_attr_level("num_colors")