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

feat(env): Needle in Haystack Curriculum (#355)

Browse source 
* needle haystack curriculum
This commit is contained in:
Zafir Stojanovski 2025-03-13 21:09:45 +01:00 • committed by GitHub
parent adea7a255e
commit 93793f5416
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 64 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/cognition/__init__.py
View file

@ -5,7 +5,7 @@ Cognition tasks for training reasoning capabilities.
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationCurriculum, ColorCubeRotationDataset
from .figlet_fonts import FigletFontConfig, FigletFontDataset
from .modulo_grid import ModuloGridConfig, ModuloGridDataset
from .needle_haystack import NeedleHaystackConfig, NeedleHaystackDataset
from .needle_haystack import NeedleHaystackConfig, NeedleHaystackCurriculum, NeedleHaystackDataset
from .number_sequences import NumberSequenceConfig, NumberSequenceCurriculum, NumberSequenceDataset
from .rectangle_count import RectangleCountConfig, RectangleCountCurriculum, RectangleCountDataset
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
@ -26,6 +26,7 @@ __all__ = [
"RectangleCountDataset",
"NeedleHaystackConfig",
"NeedleHaystackDataset",
"NeedleHaystackCurriculum",
"ModuloGridConfig",
"ModuloGridDataset",
]