open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-26 17:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_maze.py
View file

@ -1,6 +1,7 @@
import pytest
from reasoning_gym.games.maze import MazeConfig, MazeDataset, maze_dataset
from reasoning_gym import create_dataset
from reasoning_gym.games.maze import MazeConfig, MazeDataset
def test_maze_config_validation():
@ -38,7 +39,8 @@ def test_maze_dataset_creation():
def test_maze_dataset_items():
ds = maze_dataset(
ds = create_dataset(
"maze",
min_dist=3,
max_dist=5,
min_grid_size=5,
@ -62,7 +64,8 @@ def test_maze_shortest_path_correctness():
"""
min_dist = 4
max_dist = 8
ds = maze_dataset(
ds = create_dataset(
"maze",
min_dist=min_dist,
max_dist=max_dist,
min_grid_size=5,