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

Add Modulo Grid Task (#273)

Browse source 
* add modulo_grid dataset
* ensure the pattern is mathematical, not just spatial

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
This commit is contained in:
Rich Jones 2025-03-07 11:11:41 +01:00 • committed by GitHub
parent 07dc01ad87
commit cbfdf097a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 182 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 .modulo_grid import ModuloGridConfig, ModuloGridDataset
from .needle_haystack import NeedleHaystackConfig, NeedleHaystackDataset
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
from .rectangle_count import RectangleCountConfig, RectangleCountDataset
@ -22,4 +23,6 @@ __all__ = [
"RectangleCountDataset",
"NeedleHaystackConfig",
"NeedleHaystackDataset",
"ModuloGridConfig",
"ModuloGridDataset",
]