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

feat: Update arithmetic __init__.py with docstring and additional exports

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 22:26:51 +01:00
parent 00219aa3a3
commit ed71ac20cb
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
reasoning_gym/arithmetic/__init__.py
View file

@ -1,12 +1,22 @@
from .chain_sum import ChainSum, ChainSumConfig
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
"""
Arithmetic tasks for training reasoning capabilities:
- Basic arithmetic
- Chain sums
- Word problems
- Leg counting
"""
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset
from .chain_sum import ChainSum, ChainSumConfig, chain_sum
from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
__all__ = [
"ChainSum",
"ChainSumConfig",
"ArithmeticDataset",
"ArithmeticDatasetConfig",
"arithmetic_dataset",
"ChainSum",
"ChainSumConfig",
"chain_sum",
"LegCountingConfig",
"LegCountingDataset",
"leg_counting_dataset"