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

Rename ArithmeticDataset to BasicArithmeticDataset

Browse source
This commit is contained in:
Andreas Koepf 2025-01-24 10:31:26 +01:00
parent 4817c33b65
commit 0e9250bce0
7 changed files with 27 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

8
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -5,7 +5,7 @@ from ..dataset import ProceduralDataset
@dataclass
class ArithmeticDatasetConfig:
class BasicArithmeticDatasetConfig:
"""Configuration for arithmetic dataset generation"""
min_terms: int = 2
@ -34,7 +34,7 @@ class ArithmeticDatasetConfig:
class BasicArithmeticDataset(ProceduralDataset):
"""Dataset that generates basic arithmetic tasks with configurable complexity"""
def __init__(self, config: ArithmeticDatasetConfig):
def __init__(self, config: BasicArithmeticDatasetConfig):
self.config = config
self.config.validate()
super().__init__(seed=config.seed, size=config.size)
@ -183,9 +183,9 @@ def basic_arithmetic_dataset(
format_style: Style of question formatting ("simple" or "natural")
Returns:
ArithmeticDataset: Configured dataset instance
BasicArithmeticDataset: Configured dataset instance
"""
config = ArithmeticDatasetConfig(
config = BasicArithmeticDatasetConfig(
min_terms=min_terms,
max_terms=max_terms,
min_digits=min_digits,