mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
3f6b2fc807
* feat: Add Coach and ScoreBoard classes for performance tracking and difficulty adjustment * feat: Add GroupedScores class to wrap aggregated scores * refactor: Create ScoreStats class with tuple-based score statistics * feat: Add unit test for Coach with CompositeDataset and multiple datasets * fix: Add difficulty metadata to leg counting dataset * feat: Add clear() method to ScoreBoard to reset all stored data * feat: Add __len__ method to ScoreBoard to return number of scores * feat: Add update_dataset_config method to CompositeDataset * cleanup __init__ & imports
22 lines
660 B
Python
22 lines
660 B
Python
"""
|
|
Cognition tasks for training reasoning capabilities.
|
|
"""
|
|
|
|
from .arc_1d import Arc1DConfig, Arc1DDataset
|
|
from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset
|
|
from .figlet_fonts import FigletFontConfig, FigletFontDataset
|
|
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
|
|
from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
|
|
|
|
__all__ = [
|
|
"Arc1DConfig",
|
|
"Arc1DDataset",
|
|
"ColorCubeRotationConfig",
|
|
"ColorCubeRotationDataset",
|
|
"FigletFontConfig",
|
|
"FigletFontDataset",
|
|
"NumberSequenceConfig",
|
|
"NumberSequenceDataset",
|
|
"RubiksCubeConfig",
|
|
"RubiksCubeDataset",
|
|
]
|