mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2025-10-09 13:40:09 +03:00
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
"""
|
|
Cognition tasks for training reasoning capabilities.
|
|
"""
|
|
|
|
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationCurriculum, ColorCubeRotationDataset
|
|
from .figlet_fonts import FigletFontConfig, FigletFontDataset
|
|
from .modulo_grid import ModuloGridConfig, ModuloGridDataset
|
|
from .needle_haystack import NeedleHaystackConfig, NeedleHaystackCurriculum, NeedleHaystackDataset
|
|
from .number_sequences import NumberSequenceConfig, NumberSequenceCurriculum, NumberSequenceDataset
|
|
from .rectangle_count import RectangleCountConfig, RectangleCountCurriculum, RectangleCountDataset
|
|
from .rubiks_cube import RubiksCubeConfig, RubiksCubeCurriculum, RubiksCubeDataset
|
|
|
|
__all__ = [
|
|
"ColorCubeRotationConfig",
|
|
"ColorCubeRotationDataset",
|
|
"ColorCubeRotationCurriculum",
|
|
"FigletFontConfig",
|
|
"FigletFontDataset",
|
|
"NumberSequenceConfig",
|
|
"NumberSequenceDataset",
|
|
"NumberSequenceCurriculum",
|
|
"RubiksCubeConfig",
|
|
"RubiksCubeDataset",
|
|
"RubiksCubeCurriculum",
|
|
"RectangleCountConfig",
|
|
"RectangleCountCurriculum",
|
|
"RectangleCountDataset",
|
|
"NeedleHaystackConfig",
|
|
"NeedleHaystackDataset",
|
|
"NeedleHaystackCurriculum",
|
|
"ModuloGridConfig",
|
|
"ModuloGridDataset",
|
|
]
|