rlberry.envs.atari_make

rlberry.envs.atari_make(id, seed=None, **kwargs)[source]

Adaptor to manage Atari Env

Parameters:
idstr

Environment id.

seedint

seed for the environment

**kwargskeywords arguments

Optional arguments to configure the environment. (render_mode, n_frame_stack, and other arguments for StableBaselines’s make_atari_env : https://stable-baselines3.readthedocs.io/en/master/common/env_util.html#stable_baselines3.common.env_util.make_atari_env )

Returns
——-
Atari env with wrapper to be used as Gymnasium env.
Examples
——–
>>> from rlberry.envs.gym_make import atari_make
>>> env_ctor = atari_make
>>> env_kwargs = {“id”: “ALE/Freeway-v5”, “atari_wrappers_dict”:dict(terminal_on_life_loss=False),”n_frame_stack”:5}}
>>> env = env_ctor(**env_kwargs)

Examples using rlberry.envs.atari_make

A demo of ATARI Freeway environment with DQNAgent

A demo of ATARI Freeway environment with DQNAgent

A demo of ATARI Atlantis environment with vectorized PPOAgent

A demo of ATARI Atlantis environment with vectorized PPOAgent

A demo of ATARI Breakout environment with vectorized PPOAgent

A demo of ATARI Breakout environment with vectorized PPOAgent