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 reasoning_gym.create_dataset({name}, ...) global factory function

Browse source
This commit is contained in:
Andreas Koepf 2025-01-25 00:58:34 +01:00
parent 0d2d8ba6a0
commit 519e411fa5
35 changed files with 133 additions and 598 deletions
Download patch file Download diff file Expand all files Collapse all files

24
reasoning_gym/arithmetic/__init__.py
View file

@ -6,17 +6,13 @@ Arithmetic tasks for training reasoning capabilities:
- Leg counting
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig, basic_arithmetic_dataset
from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset
from .fraction_simplification import (
FractionSimplificationConfig,
FractionSimplificationDataset,
fraction_simplification_dataset,
)
from .gcd import GCDConfig, GCDDataset, gcd_dataset
from .lcm import LCMConfig, LCMDataset, lcm_dataset
from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
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
__all__ = [
"BasicArithmeticDataset",
@ -24,20 +20,14 @@ __all__ = [
"basic_arithmetic_dataset",
"ChainSum",
"ChainSumConfig",
"chain_sum_dataset",
"FractionSimplificationConfig",
"FractionSimplificationDataset",
"fraction_simplification_dataset",
"GCDConfig",
"GCDDataset",
"gcd_dataset",
"LCMConfig",
"LCMDataset",
"lcm_dataset",
"LegCountingConfig",
"LegCountingDataset",
"leg_counting_dataset",
"PrimeFactorizationConfig",
"PrimeFactorizationDataset",
"prime_factorization_dataset",
]