mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
3f6b2fc807
* feat: Add Coach and ScoreBoard classes for performance tracking and difficulty adjustment * feat: Add GroupedScores class to wrap aggregated scores * refactor: Create ScoreStats class with tuple-based score statistics * feat: Add unit test for Coach with CompositeDataset and multiple datasets * fix: Add difficulty metadata to leg counting dataset * feat: Add clear() method to ScoreBoard to reset all stored data * feat: Add __len__ method to ScoreBoard to return number of scores * feat: Add update_dataset_config method to CompositeDataset * cleanup __init__ & imports
37 lines
1.3 KiB
Python
37 lines
1.3 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 .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",
|
|
]
|