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

Merge branch 'main' into koko/gsm-symbolic-task-1

Browse source
This commit is contained in:
Adefioye 2025-02-03 01:23:26 -06:00 • committed by GitHub
commit 40b5c5583c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
49 changed files with 6334 additions and 147 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reasoning_gym/arithmetic/__init__.py
View file

@ -4,9 +4,11 @@ Arithmetic tasks for training reasoning capabilities:
- Chain sums
- Word problems
- Leg counting
- Time intervals
"""
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
@ -14,6 +16,7 @@ from .lcm import LCMConfig, LCMDataset
from .leg_counting import LegCountingConfig, LegCountingDataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
from .gsm_symbolic.gsm_symbolic_datasets import GSMSymbolicDataset, GSMSymbolicDatasetConfig
from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
__all__ = [
"BasicArithmeticDataset",
@ -21,6 +24,10 @@ __all__ = [
"basic_arithmetic_dataset",
"ChainSum",
"ChainSumConfig",
"CalendarArithmeticConfig",
"CalendarArithmeticDataset",
"Weekday",
"CalendarTask",
"FractionSimplificationConfig",
"FractionSimplificationDataset",
"GCDConfig",
@ -33,4 +40,6 @@ __all__ = [
"PrimeFactorizationDataset",
"GSMSymbolicDatasetConfig",
"GSMSymbolicDataset",
"TimeIntervalsConfig",
"TimeIntervalsDataset",
]