Selaa lähdekoodia

add reset benchmark.py and fix render manual_control.py

Rodrigo Perez-Vicente 2 vuotta sitten
vanhempi
commit
4dc78fc3b6
2 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  1. 1 0
      gym_minigrid/benchmark.py
  2. 1 2
      gym_minigrid/manual_control.py

+ 1 - 0
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):

+ 1 - 2
gym_minigrid/manual_control.py

@@ -10,8 +10,7 @@ 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)