Browse Source

remove reset from init

Rodrigo Perez-Vicente 2 years ago
parent
commit
4f8a9198b2
3 changed files with 3 additions and 7 deletions
  1. 1 1
      gym_minigrid/manual_control.py
  2. 0 3
      gym_minigrid/minigrid.py
  3. 2 3
      tests/test_envs.py

+ 1 - 1
gym_minigrid/manual_control.py

@@ -33,7 +33,7 @@ def step(env, window, action):
         print("truncated!")
         reset(env, window)
     else:
-        img = env.get_full_render()
+        img = env.get_frame()
         redraw(window, img)
 
 

+ 0 - 3
gym_minigrid/minigrid.py

@@ -926,9 +926,6 @@ class MiniGridEnv(gym.Env):
 
         self.renderer = Renderer(self.render_mode, frame_rendering)
 
-        # Initialize the state
-        self.reset()
-
     def reset(self, *, seed=None, return_info=False, options=None):
         super().reset(seed=seed)
 

+ 2 - 3
tests/test_envs.py

@@ -121,12 +121,11 @@ def test_agent_sees_method(env_id):
     env = gym.make(env_id, new_step_api=True)
     goal_pos = (env.grid.width - 2, env.grid.height - 2)
 
+    # Test the env.agent_sees() function
+    env.reset()
     # Test the "in" operator on grid objects
     assert ("green", "goal") in env.grid
     assert ("blue", "key") not in env.grid
-
-    # Test the env.agent_sees() function
-    env.reset()
     for i in range(0, 500):
         action = env.action_space.sample()
         obs, reward, terminated, truncated, info = env.step(action)