Example #1
0
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
Example #2
0
def make_env(wrap=True):
    env = gym.env.make(FLAGS.env)

    env = env.env

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