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

BF Curricula and More (#309)

Browse source 
* bf curricula
* modulo grid curricula
* minor changes to how difficulty is stored

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
This commit is contained in:
Rich Jones 2025-03-09 18:22:22 +01:00 • committed by GitHub
parent 7c7c783883
commit 46013e4640
5 changed files with 122 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_modulo_grid.py
View file

@ -1,6 +1,6 @@
import pytest
from reasoning_gym.cognition.modulo_grid import ModuloGridConfig, ModuloGridDataset
from reasoning_gym.cognition.modulo_grid import ModuloGridConfig, ModuloGridCurriculum, ModuloGridDataset
def test_modulo_grid():
@ -35,3 +35,17 @@ def test_modulo_grid():
assert item["question"] != item["answer"]
assert dataset.score_answer(answer=item["answer"], entry=item) == 1.0
assert dataset.score_answer(answer=None, entry=item) == 0.0
def test_mg_curriculum():
curriculum = ModuloGridCurriculum()
base_value = {"size": 150, "seed": 1}
base_cfg: ModuloGridConfig = curriculum.generate_configuration(base_value)
assert base_cfg.seed == 1
assert base_cfg.size == 150
assert base_cfg.size_x == 20
curriculum.increment_attr_level("size_x")
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.size_x == 30