mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-25 17:10:51 +00:00
dced3bfc45
* init * fix tests * unify codeio * filtered for libraries not present in reasoning-gym * fix more bounds * puzzle24 * knight swap curriculum * fix number sorting * fix attributes * add validation of config in creation of dataset * dry run for instantiating and validating the datasets * remove unused imports * fix curriculum tests to reference newly updated attribute names
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""
|
|
Game tasks for training reasoning capabilities:
|
|
- Board games
|
|
- Puzzle games
|
|
- Strategy games
|
|
- Simulation games
|
|
"""
|
|
|
|
from .boxnet import BoxnetConfig, BoxnetCurriculum, BoxnetDataset
|
|
from .countdown import CountdownConfig, CountdownCurriculum, CountdownDataset
|
|
from .emoji_mystery import EmojiMysteryConfig, EmojiMysteryCurriculum, EmojiMysteryDataset
|
|
from .futoshiki import FutoshikiConfig, FutoshikiCurriculum, FutoshikiDataset
|
|
from .knight_swap import KnightSwapConfig, KnightSwapCurriculum, KnightSwapDataset
|
|
from .mahjong import MahjongPuzzleConfig, MahjongPuzzleCurriculum, MahjongPuzzleDataset
|
|
from .maze import MazeConfig, MazeCurriculum, MazeDataset
|
|
from .mini_sudoku import MiniSudokuConfig, MiniSudokuCurriculum, MiniSudokuDataset
|
|
from .n_queens import NQueensConfig, NQueensCurriculum, NQueensDataset
|
|
from .puzzle24 import Puzzle24Config, Puzzle24Curriculum, Puzzle24Dataset
|
|
from .rush_hour import RushHourConfig, RushHourCurriculum, RushHourDataset
|
|
from .sokoban import SokobanConfig, SokobanCurriculum, SokobanDataset
|
|
from .sudoku import SudokuConfig, SudokuCurriculum, SudokuDataset
|
|
from .tower_of_hanoi import HanoiConfig, HanoiCurriculum, HanoiDataset
|
|
from .tsumego import TsumegoConfig, TsumegoCurriculum, TsumegoDataset
|
|
|
|
__all__ = [
|
|
"BoxnetConfig",
|
|
"BoxnetDataset",
|
|
"BoxnetCurriculum",
|
|
"CountdownConfig",
|
|
"CountdownDataset",
|
|
"CountdownCurriculum",
|
|
"EmojiMysteryConfig",
|
|
"EmojiMysteryCurriculum",
|
|
"EmojiMysteryDataset",
|
|
"FutoshikiConfig",
|
|
"FutoshikiCurriculum",
|
|
"FutoshikiDataset",
|
|
"MiniSudokuConfig",
|
|
"MiniSudokuDataset",
|
|
"MiniSudokuCurriculum",
|
|
"Puzzle24Config",
|
|
"Puzzle24Dataset",
|
|
"Puzzle24Curriculum",
|
|
"SudokuConfig",
|
|
"SudokuCurriculum",
|
|
"SudokuDataset",
|
|
"SokobanConfig",
|
|
"SokobanCurriculum",
|
|
"SokobanDataset",
|
|
"RushHourConfig",
|
|
"RushHourCurriculum",
|
|
"RushHourDataset",
|
|
"MazeConfig",
|
|
"MazeDataset",
|
|
"MazeCurriculum",
|
|
"HanoiConfig",
|
|
"HanoiDataset",
|
|
"HanoiCurriculum",
|
|
"NQueensDataset",
|
|
"NQueensConfig",
|
|
"NQueensCurriculum",
|
|
"TsumegoConfig",
|
|
"TsumegoCurriculum",
|
|
"TsumegoDataset",
|
|
"KnightSwapConfig",
|
|
"KnightSwapDataset",
|
|
"KnightSwapCurriculum",
|
|
"MahjongPuzzleConfig",
|
|
"MahjongPuzzleDataset",
|
|
"MahjongPuzzleCurriculum",
|
|
]
|