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

fix(envs): Add source dataset and index to metadata (#388)

Browse source 
* add source dataset and index to metadata

* fix typo

* fix coach class and its test
This commit is contained in:
Zafir Stojanovski 2025-03-20 12:12:14 +01:00 • committed by GitHub
parent c6d01541aa
commit 4c47527130
104 changed files with 549 additions and 146 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/test_coaching.py
View file

@ -56,12 +56,7 @@ def test_coach_with_chain_sum():
# Each inner tuple should be (param_name, value) or (param_name, (min_value, max_value))
for param in key:
assert isinstance(param, tuple)
assert param[0] in ("num_terms", "num_digits")
assert (
isinstance(param[1], int)
or (isinstance(param[1], tuple) and len(param[1]) == 2)
and all(isinstance(v, int) for v in param[1])
)
assert param[0] in ("source", "idx", "num_terms", "num_digits")
# Test aggregation with last_n
last_3 = coach.score_board.aggregate(last_n=3)