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

minor formatting changes

Browse source
This commit is contained in:
Andreas Koepf 2025-02-17 18:20:18 +01:00
parent d3323b1a77
commit aa794253fe
2 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -65,7 +65,7 @@ class BasicArithmeticDataset(ProceduralDataset):
def __init__(self, config: BasicArithmeticDatasetConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
self.added_instruction = (
"Ensure to report the answer as an integer. Please do not add commas to the integer answers reported."
" Ensure to report the answer as an integer. Do not add commas to the integer answers reported."
)
def __getitem__(self, idx: int) -> dict[str, Any]:
@ -226,15 +226,14 @@ class BasicArithmeticDataset(ProceduralDataset):
return expression, result
def _format_question(self, rng: Random, expression: str) -> str:
"""Format the expression with clear answer positioning"""
# answer_instruction = "Put your final answer after '=' without additional text."
"""Format the the question with the arithmetic expression"""
if self.config.format_style == "simple":
return f"Calculate {expression}. "
return f"Calculate {expression}."
else:
templates = ["What is {0}. ", "Solve {0}. ", "Compute {0}. ", "Evaluate: {0}. "]
template = rng.choice(templates).format(expression)
return f"{template}"
templates = ["What is {0}?", "Solve {0}.", "Compute {0}.", "Evaluate: {0}."]
template = rng.choice(templates)
return template.format(expression)
# Register the dataset