Farama-Foundation / Arcade-Learning-Environment

The Arcade Learning Environment (ALE) -- a platform for AI research.
GNU General Public License v2.0
2.12k stars 420 forks source link

FIX: add render_mode attribute #476

Closed younik closed 1 year ago

younik commented 1 year ago

This PR addresses https://github.com/mgbellemare/Arcade-Learning-Environment/issues/471 and https://github.com/mgbellemare/Arcade-Learning-Environment/issues/475 by adding the property render_mode to AtariEnv.

Quick test:

import gym
from gym.wrappers import RecordVideo

env = RecordVideo(
    gym.make('ALE/Breakout-v5', render_mode="rgb_array"),
    video_folder='video'
)

env.reset()
for _ in range(100):
    env.step(env.action_space.sample())

env.close()

@JesseFarebro

JesseFarebro commented 1 year ago

Thanks for fixing this! 🙌