瀏覽代碼

Fixed bug in place_agent

Maxime Chevalier-Boisvert 7 年之前
父節點
當前提交
342b3c96f1
共有 2 個文件被更改,包括 5 次插入0 次删除
  1. 1 0
      gym_minigrid/minigrid.py
  2. 4 0
      run_tests.py

+ 1 - 0
gym_minigrid/minigrid.py

@@ -869,6 +869,7 @@ class MiniGridEnv(gym.Env):
         Set the agent's starting point at an empty position in the grid
         """
 
+        self.start_pos = None
         pos = self.place_obj(None, top, size)
         self.start_pos = pos
 

+ 4 - 0
run_tests.py

@@ -43,6 +43,10 @@ for envName in env_list:
 
         obs, reward, done, info = env.step(action)
 
+        # Validate the agent position
+        assert env.agent_pos[0] < env.grid_size
+        assert env.agent_pos[1] < env.grid_size
+
         # Test observation encode/decode roundtrip
         img = obs['image']
         grid = Grid.decode(img)