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

corrected failing airthmetic test

Browse source
This commit is contained in:
joesharratt1229 2025-02-16 12:01:54 +00:00
parent a59e4cc918
commit 569517664f
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -227,15 +227,16 @@ class BasicArithmeticDataset(ProceduralDataset):
answer_instruction = "Put your final answer after '=' without additional text."
if self.config.format_style == "simple":
return f"Calculate {expression} ="
return f"{answer_instruction} Calculate {expression} ="
else:
templates = [
"What is {0}? =",
"Solve {0} and write answer after =",
"What is {0} =",
"Solve {0}=",
"Compute {0} =",
"Evaluate: {0} ="
]
return rng.choice(templates).format(expression) + f" {answer_instruction}"
template = rng.choice(templates).format(expression)
return f"{answer_instruction} {template}"
# Register the dataset