reasoning-gym/reasoning_gym/cognition/__init__.py
2025-02-08 19:37:26 +01:00

19 lines
575 B
Python

"""
Cognition tasks for training reasoning capabilities.
"""
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset
from .figlet_fonts import FigletFontConfig, FigletFontDataset
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
__all__ = [
"ColorCubeRotationConfig",
"ColorCubeRotationDataset",
"FigletFontConfig",
"FigletFontDataset",
"NumberSequenceConfig",
"NumberSequenceDataset",
"RubiksCubeConfig",
"RubiksCubeDataset",
]