Skip to content

Commit

Permalink
Fix: (In dataset creation tutorial) Register PointMaze_Medium-v3 and …
Browse files Browse the repository at this point in the history
…Correct env.maze Access (#272)

Co-authored-by: Omar Younis <[email protected]>
  • Loading branch information
mengyuest and younik authored Jan 6, 2025
1 parent 3c7b0fa commit e482bf5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion docs/tutorials/dataset_creation/point_maze_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit e482bf5

Please sign in to comment.