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

adds Needle in a Haystack problems

Browse source
This commit is contained in:
Rich Jones 2025-02-20 12:28:30 +01:00
parent 8b03bf6a92
commit 621c20d8d8
4 changed files with 3584 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/cognition/__init__.py
View file

@ -4,6 +4,7 @@ Cognition tasks for training reasoning capabilities.
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset
from .figlet_fonts import FigletFontConfig, FigletFontDataset
from .needle_haystack import NeedleHaystackConfig, NeedleHaystackDataset
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
from .rectangle_count import RectangleCountConfig, RectangleCountDataset
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
@ -19,4 +20,6 @@ __all__ = [
"RubiksCubeDataset",
"RectangleCountConfig",
"RectangleCountDataset",
"NeedleHaystackConfig",
"NeedleHaystackDataset",
]