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

adapt needle_haystack test to curriculum changes

Browse source
This commit is contained in:
Andreas Koepf 2025-03-16 17:35:16 +01:00
parent f25820a0c7
commit c760da6b1b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_needle_haystack.py
View file

@ -89,7 +89,7 @@ def test_needle_haystack_curriculum():
# test incrementing attribute levels
curriculum.increment_attr_level("num_statements")
increased_cfg = curriculum.generate_configuration(base_value)
assert increased_cfg.min_num_statements == 10 and increased_cfg.max_num_statements == 1000
assert increased_cfg.min_num_statements == 10 and increased_cfg.max_num_statements == 500
# test decrementing attribute level for num_statements again
curriculum.decrement_attr_level("num_statements")