garage.envs.wrappers.atari_env module

Episodic life wrapper for gym.Env.

class AtariEnv(env)

Bases: gym.core.Wrapper

Atari environment wrapper for gym.Env.

This wrapper convert the observations returned from baselines wrapped environment, which is a LazyFrames object into numpy arrays.

Parameters:env (gym.Env) – The environment to be wrapped.

reset(**kwargs)

gym.Env reset function.

step(action)

gym.Env step function.