add reasoning_gym.create_dataset({name}, ...) global factory function

This commit is contained in:
Andreas Koepf 2025-01-25 00:58:34 +01:00
parent 0d2d8ba6a0
commit 519e411fa5
35 changed files with 133 additions and 598 deletions

View file

@ -5,18 +5,15 @@ Game tasks for training reasoning capabilities:
- Strategy games
"""
from .maze import MazeConfig, MazeDataset, maze_dataset
from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset, mini_sudoku_dataset
from .sudoku import SudokuConfig, SudokuDataset, sudoku_dataset
from .maze import MazeConfig, MazeDataset
from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset
from .sudoku import SudokuConfig, SudokuDataset
__all__ = [
"MiniSudokuConfig",
"MiniSudokuDataset",
"mini_sudoku_dataset",
"SudokuConfig",
"SudokuDataset",
"sudoku_dataset",
"MazeConfig",
"MazeDataset",
"maze_dataset",
]