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

add reasoning_gym.create_dataset({name}, ...) global factory function

Browse source
This commit is contained in:
Andreas Koepf 2025-01-25 00:58:34 +01:00
parent e9549f2a63
commit 0dcff77b37
35 changed files with 133 additions and 598 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/cognition/__init__.py
View file

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