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

Add QL puzz

Browse source
This commit is contained in:
Rich Jones 2025-01-29 23:33:39 +01:00
parent 9f9359b3e4
commit d230b5da97
2 changed files with 53 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_quantum_lock.py
View file

@ -5,7 +5,7 @@ from reasoning_gym.graphs.quantum_lock import QuantumLockConfig, QuantumLockData
def test_quantumlock_items():
"""Test basic properties and solution of generated items"""
config = QuantumLockConfig(
difficulty=3,
difficulty=10,
)
dataset = QuantumLockDataset(config)
@ -19,8 +19,5 @@ def test_quantumlock_items():
assert "solution_path" in item["metadata"]
assert "difficulty" in item["metadata"]
print(item['question'])
print(item['answer'])
assert dataset.score_answer(answer=item['metadata']['solution_path'], entry=item) == 1.0
assert dataset.score_answer(answer=None, entry=item) == 0.0