open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add time interval dataset class

Browse source
This commit is contained in:
Andreas Koepf 2025-02-01 02:10:48 +01:00
parent 6e9f20568e
commit 44f32e3862
5 changed files with 446 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/__init__.py
View file

@ -4,6 +4,7 @@ Arithmetic tasks for training reasoning capabilities:
- Chain sums
- Word problems
- Leg counting
- Time intervals
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
@ -13,6 +14,7 @@ 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",
@ -30,4 +32,6 @@ __all__ = [
"LegCountingDataset",
"PrimeFactorizationConfig",
"PrimeFactorizationDataset",
"TimeIntervalsConfig",
"TimeIntervalsDataset",
]