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

refactor: Rename ArithmeticDataset to BasicArithmeticDataset

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 10:02:02 +01:00
parent 1b556b70f7
commit 95d9974911
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

12
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -31,8 +31,8 @@ class ArithmeticDatasetConfig:
assert op in ["+", "-", "*"], f"unsupported operator: {op}"
class ArithmeticDataset(ProceduralDataset):
"""Dataset that generates arithmetic tasks with configurable complexity"""
class BasicArithmeticDataset(ProceduralDataset):
"""Dataset that generates basic arithmetic tasks with configurable complexity"""
def __init__(self, config: ArithmeticDatasetConfig):
self.config = config
@ -155,7 +155,7 @@ class ArithmeticDataset(ProceduralDataset):
return rng.choice(templates).format(expression)
def arithmetic_dataset(
def basic_arithmetic_dataset(
min_terms: int = 2,
max_terms: int = 6,
min_digits: int = 1,
@ -167,8 +167,8 @@ def arithmetic_dataset(
size: int = 500,
format_style: Literal["simple", "natural"] = "simple",
whitespace: Literal["no_space", "single", "random"] = "single",
) -> ArithmeticDataset:
"""Create an ArithmeticDataset with the given configuration.
) -> BasicArithmeticDataset:
"""Create a BasicArithmeticDataset with the given configuration.
Args:
min_terms: Minimum number of terms in expressions
@ -198,4 +198,4 @@ def arithmetic_dataset(
format_style=format_style,
whitespace=whitespace,
)
return ArithmeticDataset(config)
return BasicArithmeticDataset(config)