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

add rectangle count dataset

Browse source
This commit is contained in:
Rich Jones 2025-02-11 13:56:27 +01:00
parent a174f44361
commit 2fa1ea106d
3 changed files with 157 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/cognition/__init__.py
View file

@ -5,6 +5,7 @@ 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 .rectangle_count import RectangleCountConfig, RectangleCountDataset
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
__all__ = [
@ -16,4 +17,6 @@ __all__ = [
"NumberSequenceDataset",
"RubiksCubeConfig",
"RubiksCubeDataset",
"RectangleCountConfig",
"RectangleCountDataset",
]