init definitions

This commit is contained in:
Rich Jones 2025-01-30 17:15:48 +01:00
parent 2393ae0525
commit 9d4f896329
3 changed files with 20 additions and 1 deletions

View file

@ -8,10 +8,16 @@ Cognition tasks for training reasoning capabilities:
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
from .figlet_fonts import FigletFontConfig, FigletFontDataset
__all__ = [
"NumberSequenceConfig",
"NumberSequenceDataset",
"ColorCubeRotationConfig",
"ColorCubeRotationDataset",
"RubiksCubeConfig",
"RubiksCubeDataset",
"FigletFontConfig",
"FigletFontDataset"
]