@@ -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):
@@ -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)