Преглед изворни кода

fix typo + remove unused package

saleml пре 2 година
родитељ
комит
57fa6e5fe1
2 измењених фајлова са 5 додато и 4 уклоњено
  1. 0 1
      gym_minigrid/envs/dynamicobstacles.py
  2. 5 3
      gym_minigrid/envs/obstructedmaze.py

+ 0 - 1
gym_minigrid/envs/dynamicobstacles.py

@@ -1,6 +1,5 @@
 from operator import add
 
-import gym
 from gym.spaces import Discrete
 
 from gym_minigrid.minigrid import Ball, Goal, Grid, MiniGridEnv

+ 5 - 3
gym_minigrid/envs/obstructedmaze.py

@@ -20,7 +20,7 @@ class ObstructedMazeEnv(RoomGrid):
             max_steps=max_steps,
             **kwargs
         )
-        self.obj = Ball()  # intiale the obj attribute, that will be changed later on
+        self.obj = Ball()  # initialize the obj attribute, that will be changed later on
 
     def _gen_grid(self, width, height):
         super()._gen_grid(width, height)
@@ -67,7 +67,8 @@ class ObstructedMazeEnv(RoomGrid):
         if blocked:
             vec = DIR_TO_VEC[door_idx]
             blocking_ball = Ball(self.blocking_ball_color) if blocked else None
-            self.grid.set(door_pos[0] - vec[0], door_pos[1] - vec[1], blocking_ball)
+            self.grid.set(door_pos[0] - vec[0],
+                          door_pos[1] - vec[1], blocking_ball)
 
         if locked:
             obj = Key(door.color)
@@ -105,7 +106,8 @@ class ObstructedMaze_1Dlhb(ObstructedMazeEnv):
             blocked=self.blocked,
         )
 
-        self.obj, _ = self.add_object(1, 0, "ball", color=self.ball_to_find_color)
+        self.obj, _ = self.add_object(
+            1, 0, "ball", color=self.ball_to_find_color)
         self.place_agent(0, 0)