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