From e482bf5bbc2b8ad4025f2f10c57fcaa09955f5bf Mon Sep 17 00:00:00 2001 From: Yue Meng Date: Mon, 6 Jan 2025 05:50:33 -0500 Subject: [PATCH] Fix: (In dataset creation tutorial) Register PointMaze_Medium-v3 and Correct env.maze Access (#272) Co-authored-by: Omar Younis --- docs/tutorials/dataset_creation/point_maze_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/tutorials/dataset_creation/point_maze_dataset.py b/docs/tutorials/dataset_creation/point_maze_dataset.py index a3f7f2d7..b72558c1 100644 --- a/docs/tutorials/dataset_creation/point_maze_dataset.py +++ b/docs/tutorials/dataset_creation/point_maze_dataset.py @@ -20,6 +20,7 @@ # Lets start by importing the required modules for this tutorial: import gymnasium as gym +import gymnasium_robotics # noqa: F401 import numpy as np from minari import DataCollector, StepDataCallback @@ -383,7 +384,7 @@ def __call__( obs, _ = collector_env.reset(seed=123) -waypoint_controller = WaypointController(maze=env.maze) +waypoint_controller = WaypointController(maze=env.unwrapped.maze) for n_step in range(int(total_steps)): action = waypoint_controller.compute_action(obs)