Add Maze dataset and config

This commit is contained in:
panispani 2025-01-24 21:56:40 +00:00
parent f961cdbfd7
commit 4032b2de23
2 changed files with 207 additions and 0 deletions

View file

@ -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",
]