Merge pull request #240 from Farama-Foundation/fix-mazeant-import

Fix MazeAnt-v5 environments having incorect entry points
Farama-Foundation · Oct 11, 2024 · 2d3e0df · 2d3e0df
2 parents 3e29629 + 61f2af3
commit 2d3e0df
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/gymnasium_robotics/__init__.py b/gymnasium_robotics/__init__.py
@@ -850,7 +850,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Open{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.OPEN,
@@ -862,7 +862,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Open_Diverse_G{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.OPEN_DIVERSE_G,
@@ -874,7 +874,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Open_Diverse_GR{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.OPEN_DIVERSE_GR,
@@ -886,7 +886,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Medium{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.MEDIUM_MAZE,
@@ -898,7 +898,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Medium_Diverse_G{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.MEDIUM_MAZE_DIVERSE_G,
@@ -910,7 +910,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Medium_Diverse_GR{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.MEDIUM_MAZE_DIVERSE_GR,
@@ -922,7 +922,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Large{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.LARGE_MAZE,
@@ -934,7 +934,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Large_Diverse_G{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.LARGE_MAZE_DIVERSE_G,
@@ -946,7 +946,7 @@ def _merge(a, b):
 
             register(
                 id=f"AntMaze_Large_Diverse_GR{suffix}-{version}",
-                entry_point="gymnasium_robotics.envs.maze.ant_maze:AntMazeEnv",
+                entry_point=f"gymnasium_robotics.envs.maze.ant_maze_{version}:AntMazeEnv",
                 kwargs=_merge(
                     {
                         "maze_map": maps.LARGE_MAZE_DIVERSE_GR,