open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-22 16:49:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert merge conflicts

Browse source
This commit is contained in:
EduardDurech 2025-02-07 11:38:17 +00:00
parent 4c3ae0aebf
commit 227319f1da
2 changed files with 40 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

48
reasoning_gym/arithmetic/__init__.py
View file

@ -7,24 +7,36 @@ Arithmetic tasks for training reasoning capabilities:
- 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
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 .lcm import LCMConfig, LCMDataset
from .leg_counting import LegCountingConfig, LegCountingDataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
__all__ = [
# "BasicArithmeticDataset",
# "CalendarArithmeticDataset",
"ChainSumDataset",
# "FractionSimplificationDataset",
# "GcdDataset",
# "LcmDataset",
# "LegCountingDataset",
# "PrimeFactorizationDataset",
# "TimeIntervalsDataset",
"BasicArithmeticDataset",
"BasicArithmeticDatasetConfig",
"basic_arithmetic_dataset",
"ChainSum",
"ChainSumConfig",
"CalendarArithmeticConfig",
"CalendarArithmeticDataset",
"Weekday",
"CalendarTask",
"FractionSimplificationConfig",
"FractionSimplificationDataset",
"GCDConfig",
"GCDDataset",
"LCMConfig",
"LCMDataset",
"LegCountingConfig",
"LegCountingDataset",
"PrimeFactorizationConfig",
"PrimeFactorizationDataset",
"TimeIntervalsConfig",
"TimeIntervalsDataset",
]