Template Abstraction

reasoning_gym/exercises/arithmetic/__init__.py

@@ -0,0 +1,30 @@
+"""
+Arithmetic tasks for training reasoning capabilities:
+- Basic arithmetic
+- Chain sums
+- Word problems
+- Leg counting
+- Time intervals
+"""
+
+# from .basic_arithmetic import BasicArithmeticDataset
+# from .calendar_arithmetic import CalendarArithmeticDataset
+from .chain_sum import ChainSumExercise
+# from .fraction_simplification import FractionSimplificationDataset
+# from .gcd import GcdDataset
+# from .lcm import LcmDataset
+# from .leg_counting import LegCountingDataset
+# from .prime_factorization import PrimeFactorizationDataset
+# from .time_intervals import TimeIntervalsDataset
+
+__all__ = [
+    # "BasicArithmeticDataset",
+    # "CalendarArithmeticDataset",
+    "ChainSumExercise",
+    # "FractionSimplificationDataset",
+    # "GcdDataset",
+    # "LcmDataset",
+    # "LegCountingDataset",
+    # "PrimeFactorizationDataset",
+    # "TimeIntervalsDataset",
+]