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

feat: Add SimpleEquationsDataset to algebra module

Browse source 
Developed together with Benjamin Rio (benjamrio)
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 15:50:27 +01:00 • committed by Andreas Koepf
parent 853777c290
commit b66b1c09b9
5 changed files with 59 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

16
reasoning_gym/algebra/test.py
View file

@ -16,8 +16,22 @@ def test_simple_equations_generation():
# Validate equation format
equation = item["metadata"]["equation"]
variable = item["metadata"]["variable"]
assert "=" in equation
assert item["metadata"]["variable"] in equation
assert variable in equation
# Validate question format
question = item["question"]
assert variable in question
assert equation in question
assert any(
prompt in question
for prompt in [
"Find the value of",
"Solve for",
"Determine the value of"
]
)
def test_simple_equations_config():