Bläddra i källkod

use the new render mode api

saleml 2 år sedan
förälder
incheckning
8ccd07a1ec
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      manual_control.py

+ 2 - 2
manual_control.py

@@ -10,7 +10,7 @@ from gym_minigrid.window import Window
 
 def redraw(img):
     if not args.agent_view:
-        img = env.render('rgb_array', tile_size=args.tile_size)
+        img = env.render(tile_size=args.tile_size)
 
     window.show_img(img)
 
@@ -98,7 +98,7 @@ parser.add_argument(
 
 args = parser.parse_args()
 
-env = gym.make(args.env)
+env = gym.make(args.env, render_mode='rgb_array')
 
 if args.agent_view:
     env = RGBImgPartialObsWrapper(env)