add to init

reasoning_gym/arithmetic/__init__.py

@@ -6,6 +6,7 @@ from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConf
 from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
 from .chain_sum import ChainSumConfig, ChainSumDataset
 from .count_bits import CountBitsConfig, CountBitsDataset
+from .decimal_arithmetic import DecimalArithmeticConfig, DecimalArithmeticDataset
 from .decimal_chain_sum import DecimalChainSumConfig, DecimalChainSumDataset
 from .dice import DiceConfig, DiceDataset
 from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset
@@ -50,4 +51,8 @@ __all__ = [
     "DiceDataset",
     "NumberFormatConfig",
     "NumberFormatDataset",
+    "DecimalArithmeticConfig",
+    "DecimalArithmeticDataset",
+    "DecimalChainSumConfig",
+    "DecimalChainSumDataset",
 ]

reasoning_gym/arithmetic/decimal_arithmetic.py

@@ -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)