open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Andreas Koepf 2025-01-24 17:39:37 +01:00
parent 22ec380a98
commit 49ff0742b8
5 changed files with 46 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

12
reasoning_gym/cognition/__init__.py
View file

@ -6,6 +6,14 @@ Cognition tasks for training reasoning capabilities:
- Working memory
"""
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset, sequence_dataset
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset, color_cube_rotation_dataset
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset, number_sequence_dataset
__all__ = ["NumberSequenceDataset", "NumberSequenceConfig", "sequence_dataset"]
__all__ = [
"NumberSequenceConfig",
"NumberSequenceDataset",
"number_sequence_dataset",
"ColorCubeRotationConfig",
"ColorCubeRotationDataset",
"color_cube_rotation_dataset",
]