refactor: Rename ArithmeticDataset to BasicArithmeticDataset

reasoning_gym/arithmetic/__init__.py

@@ -6,7 +6,7 @@ Arithmetic tasks for training reasoning capabilities:
 - Leg counting
 """
 
-from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset
+from .basic_arithmetic import BasicArithmeticDataset, ArithmeticDatasetConfig, basic_arithmetic_dataset
 from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset
 from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset, fraction_simplification_dataset
 from .gcd import GCDConfig, GCDDataset, gcd_dataset
@@ -15,9 +15,9 @@ from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_da
 from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset
 
 __all__ = [
-    "ArithmeticDataset",
+    "BasicArithmeticDataset",
     "ArithmeticDatasetConfig",
-    "arithmetic_dataset",
+    "basic_arithmetic_dataset",
     "ChainSum",
     "ChainSumConfig",
     "chain_sum_dataset",

reasoning_gym/arithmetic/basic_arithmetic.py

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