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

include ranges rather than sampled values in difficulty metadata dicts (#387)

Browse source 
* update difficulty metadata for logic datasets

* update difficulty metadata for graph datasets

* update difficulty metadata for geometry datasets

* update difficulty metadata for games datasets

* update difficulty metadata for cognition datasets

* update difficulty metadata for arithmetic datasets

* update difficulty metadata for arc datasets

* update difficulty metadata for algorithmic datasets

* update difficulty metadata for algebra datasets

* use tuples

* update tests

* update tests
This commit is contained in:
Oliver Stanley 2025-03-20 09:27:03 +00:00 • committed by GitHub
parent b69c35818a
commit 7475a20700
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
80 changed files with 304 additions and 126 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_boxnet.py
View file

@ -43,8 +43,8 @@ def test_boxnet_items():
assert "initial_state" in item["metadata"]
# Verify row_num and column_num are within limits
row_num = item["metadata"]["difficulty"]["row_num"]
column_num = item["metadata"]["difficulty"]["column_num"]
row_num = item["metadata"]["row_num"]
column_num = item["metadata"]["column_num"]
assert 1 <= row_num <= 2, f"row_num {row_num} outside valid range"
assert 1 <= column_num <= 2, f"column_num {column_num} outside valid range"
@ -78,8 +78,8 @@ def test_boxnet_grid_sizes():
for i in range(len(dataset)):
item = dataset[i]
row_num = item["metadata"]["difficulty"]["row_num"]
column_num = item["metadata"]["difficulty"]["column_num"]
row_num = item["metadata"]["row_num"]
column_num = item["metadata"]["column_num"]
rows_set.add(row_num)
columns_set.add(column_num)