浏览代码

use the new render mode api

saleml 2 年之前
父节点
当前提交
8ccd07a1ec
共有 1 个文件被更改,包括 2 次插入2 次删除
  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)