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

add to init

Browse source
This commit is contained in:
Rich Jones 2025-02-20 10:51:00 +01:00
parent 0f965c157f
commit da9b215278
3 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/decimal_arithmetic.py
View file

@ -8,7 +8,7 @@ from ..factory import ProceduralDataset, register_dataset
@dataclass
class DecimalArithmeticDatasetConfig:
class DecimalArithmeticConfig:
"""Configuration for decimal arithmetic dataset generation"""
min_num_decimal_places: int = 6
@ -140,7 +140,7 @@ def _eval_ast(node) -> Decimal:
class DecimalArithmeticDataset(ProceduralDataset):
"""Dataset that generates basic arithmetic tasks using Decimal arithmetic and proper operator precedence."""
def __init__(self, config: DecimalArithmeticDatasetConfig):
def __init__(self, config: DecimalArithmeticConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
def __getitem__(self, idx: int) -> dict[str, Any]:
@ -202,4 +202,4 @@ class DecimalArithmeticDataset(ProceduralDataset):
# Register the dataset with the factory.
register_dataset("decimal_arithmetic", DecimalArithmeticDataset, DecimalArithmeticDatasetConfig)
register_dataset("decimal_arithmetic", DecimalArithmeticDataset, DecimalArithmeticConfig)