mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-22 16:49:06 +00:00
Add Maze dataset and config
This commit is contained in:
parent
f961cdbfd7
commit
4032b2de23
2 changed files with 207 additions and 0 deletions
|
|
@ -7,6 +7,7 @@ Game tasks for training reasoning capabilities:
|
|||
|
||||
from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset, mini_sudoku_dataset
|
||||
from .sudoku import SudokuConfig, SudokuDataset, sudoku_dataset
|
||||
from .maze import MazeConfig, MazeDataset, maze_dataset
|
||||
|
||||
__all__ = [
|
||||
"MiniSudokuConfig",
|
||||
|
|
@ -15,4 +16,7 @@ __all__ = [
|
|||
"SudokuConfig",
|
||||
"SudokuDataset",
|
||||
"sudoku_dataset",
|
||||
"MazeConfig",
|
||||
"MazeDataset",
|
||||
"maze_dataset",
|
||||
]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue