reasoning-gym/reasoning_gym/arithmetic/__init__.py
Zafir Stojanovski a8c39ddcfb
Power Function (#102)
* power function dataset + tests
2025-02-10 22:04:58 +01:00

38 lines
1.4 KiB
Python

"""
Arithmetic tasks for training reasoning capabilities:
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
from .chain_sum import ChainSum, ChainSumConfig
from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset
from .gcd import GCDConfig, GCDDataset
from .gsm_symbolic.gsm_symbolic import GSMSymbolicDataset, GSMSymbolicDatasetConfig
from .lcm import LCMConfig, LCMDataset
from .leg_counting import LegCountingConfig, LegCountingDataset
from .power_function import PowerFunctionConfig, PowerFunctionDataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
__all__ = [
"BasicArithmeticDataset",
"BasicArithmeticDatasetConfig",
"ChainSum",
"ChainSumConfig",
"CalendarArithmeticConfig",
"CalendarArithmeticDataset",
"FractionSimplificationConfig",
"FractionSimplificationDataset",
"GCDConfig",
"GCDDataset",
"LCMConfig",
"LCMDataset",
"LegCountingConfig",
"LegCountingDataset",
"PrimeFactorizationConfig",
"PrimeFactorizationDataset",
"GSMSymbolicDatasetConfig",
"GSMSymbolicDataset",
"TimeIntervalsConfig",
"TimeIntervalsDataset",
]