Python AtariEnvWrapper Exemples

Exemple #1

def make_env(wrap=True):
    env = gym.envs.make(FLAGS.env)
    # remove the timelimitwrapper
    env = env.env
    if wrap:
        env = atari_helpers.AtariEnvWrapper(env)
    return env

Exemple #2

def make_env(wrap=True):
    env = gym.env.make(FLAGS.env)

    env = env.env

    if wrap:
        env = atari_helpers.AtariEnvWrapper(env)
    return env