mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-22 16:49:06 +00:00
19 lines
555 B
Python
19 lines
555 B
Python
"""
|
|
Cognition tasks for training reasoning capabilities:
|
|
- Pattern recognition
|
|
- Sequence completion
|
|
- Logical reasoning
|
|
- Working memory
|
|
"""
|
|
|
|
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset, color_cube_rotation_dataset
|
|
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset, number_sequence_dataset
|
|
|
|
__all__ = [
|
|
"NumberSequenceConfig",
|
|
"NumberSequenceDataset",
|
|
"number_sequence_dataset",
|
|
"ColorCubeRotationConfig",
|
|
"ColorCubeRotationDataset",
|
|
"color_cube_rotation_dataset",
|
|
]
|