Merge pull request #210 from rodrigodelazcano/old_gym_versions

gym_minigrid/benchmark.py

@@ -41,6 +41,7 @@ env = gym.make(args.env_name)
 env = RGBImgPartialObsWrapper(env)
 env = ImgObsWrapper(env)
 
+env.reset()
 # Benchmark rendering
 t0 = time.time()
 for i in range(args.num_frames):

gym_minigrid/manual_control.py

@@ -10,14 +10,12 @@ from gym_minigrid.wrappers import ImgObsWrapper, RGBImgPartialObsWrapper
 
 def redraw(img):
     if not args.agent_view:
-        img = env.render(tile_size=args.tile_size)
-
+        img = env.render(mode="rgb_array", tile_size=args.tile_size)
     window.show_img(img)
 
 
 def reset():
-    seed = None if args.seed == -1 else args.seed
-    obs = env.reset(seed=seed)
+    obs = env.reset()
 
     if hasattr(env, "mission"):
         print("Mission: %s" % env.mission)
@@ -93,7 +91,8 @@ parser.add_argument(
 
 args = parser.parse_args()
 
-env = gym.make(args.env)
+seed = None if args.seed == -1 else args.seed
+env = gym.make(args.env, seed=seed)
 
 if args.agent_view:
     env = RGBImgPartialObsWrapper(env)