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

20
reasoning_gym/algorithmic/__init__.py
View file

@ -6,31 +6,21 @@ Algorithmic tasks for training reasoning capabilities:
- Pattern matching
"""
from reasoning_gym.arithmetic.basic_arithmetic import basic_arithmetic_dataset
from reasoning_gym.arithmetic.chain_sum import chain_sum_dataset
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 .number_sorting import NumberSortingConfig, NumberSortingDataset, number_sorting_dataset
from .word_reversal import WordReversalConfig, WordReversalDataset, word_reversal_dataset
from .base_conversion import BaseConversionConfig, BaseConversionDataset
from .letter_counting import LetterCountingConfig, LetterCountingDataset
from .number_filtering import NumberFilteringConfig, NumberFilteringDataset
from .number_sorting import NumberSortingConfig, NumberSortingDataset
from .word_reversal import WordReversalConfig, WordReversalDataset
__all__ = [
"basic_arithmetic_dataset",
"BaseConversionConfig",
"BaseConversionDataset",
"base_conversion_dataset",
"chain_sum_dataset",
"LetterCountingConfig",
"LetterCountingDataset",
"letter_counting_dataset",
"NumberFilteringConfig",
"NumberFilteringDataset",
"number_filtering_dataset",
"NumberSortingConfig",
"NumberSortingDataset",
"number_sorting_dataset",
"WordReversalConfig",
"WordReversalDataset",
"word_reversal_dataset",
]