open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-22 16:49:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make task entries json serializable (#443)

Browse source 
* make sympy-based task entries json serializable

* remove datetime objs from time_intervals metadata

* make adv geometry json serializable

* make futoshiki metadata json serializable

* fixes

* futoshiki tweaks

* fix adv geometry

* deal with fractions in str representations

* fix

* restore start_time, end_time as str
This commit is contained in:
Oliver Stanley 2025-06-02 07:57:15 +01:00 • committed by GitHub
parent 6614338ecc
commit c0e98f93b4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 36 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_futoshiki.py
View file

@ -58,7 +58,7 @@ def test_futoshiki_items():
assert len(row) <= config.max_board_size
# Verify constraints format
constraints = metadata["constraints"]
for ((r1, c1), (r2, c2)), rel in constraints.items():
for r1, c1, r2, c2, rel in constraints:
assert 0 <= r1 < config.max_board_size
assert 0 <= c1 < config.max_board_size
assert 0 <= r2 < config.max_board_size
@ -97,7 +97,9 @@ def test_futoshiki_solution_validity():
item = dataset[i]
metadata = item["metadata"]
solution = metadata["solution"]
constraints = metadata["constraints"]
constraints_meta = metadata["constraints"]
constraints = {((r1, c1), (r2, c2)): rel for (r1, c1, r2, c2, rel) in constraints_meta}
assert is_valid_solution(solution, config.min_board_size, constraints)
@ -111,7 +113,9 @@ def test_futoshiki_puzzle_solvability():
item = dataset[i]
metadata = item["metadata"]
puzzle = metadata["puzzle"]
constraints = metadata["constraints"]
constraints_meta = metadata["constraints"]
constraints = {((r1, c1), (r2, c2)): rel for (r1, c1, r2, c2, rel) in constraints_meta}
# Verify puzzle has exactly one solution
assert dataset.count_solutions(puzzle, constraints, limit=2) == 1