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: Add arithmetic dataset functions to algorithmic package __init__

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

4
reasoning_gym/algorithmic/__init__.py
View file

@ -6,15 +6,19 @@ Algorithmic tasks for training reasoning capabilities:
- Pattern matching
"""
from reasoning_gym.arithmetic.basic_arithmetic import arithmetic_dataset
from reasoning_gym.arithmetic.chain_sum import chain_sum
from .base_conversion import BaseConversionConfig, BaseConversionDataset, base_conversion_dataset
from .letter_counting import LetterCountingConfig, LetterCountingDataset, letter_counting_dataset
from .number_filtering import NumberFilteringConfig, NumberFilteringDataset, number_filtering_dataset
from .word_reversal import WordReversalConfig, WordReversalDataset, word_reversal_dataset
__all__ = [
"arithmetic_dataset",
"BaseConversionConfig",
"BaseConversionDataset",
"base_conversion_dataset",
"chain_sum",
"LetterCountingConfig",
"LetterCountingDataset",
"letter_counting_dataset",