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

added power function exponent (#291)

Browse source 
* added power function exponent

* register PowerFunctionCurriculum

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
This commit is contained in:
joesharratt1229 2025-03-08 01:54:36 +01:00 • committed by GitHub
parent 88a3d065bd
commit e2e094376e
3 changed files with 46 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/test_power_function.py
View file

@ -62,3 +62,23 @@ def test_power_function_score_function():
for item in dataset:
answer = item["answer"]
assert dataset.score_answer(answer, item) == 1.0
def test_power_function_curriculum():
"""Test PowerFunctionCurriculum configuration generation and attribute manipulation"""
from reasoning_gym.arithmetic import PowerFunctionCurriculum
curriculum = PowerFunctionCurriculum()
base_value = {"size": 150, "seed": 1}
base_cfg = curriculum.generate_configuration(base_value)
assert base_cfg.seed == 1
assert base_cfg.size == 150
assert base_cfg.min_exponent == 2 and base_cfg.max_exponent == 2
# Test incrementing attribute levels for exponent & base attributes
curriculum.increment_attr_level("exponent")
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.min_exponent == 2 and increased_cfg.max_exponent == 4