reasoning-gym/reasoning_gym/arithmetic/__init__.py
2025-02-07 11:27:21 +00:00

30 lines
863 B
Python

"""
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 ChainSumDataset
# 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",
"ChainSumDataset",
# "FractionSimplificationDataset",
# "GcdDataset",
# "LcmDataset",
# "LegCountingDataset",
# "PrimeFactorizationDataset",
# "TimeIntervalsDataset",
]