def wrap_atari(env, max_episode_steps=None):
    assert ('NoFrameskip' in env.spec.id)
    env = NoopResetEnv(env, noop_max=30)
    env = MaxAndSkipEnv(env, skip=4)
    assert (max_episode_steps is None)
    return env
def wrap_atari(env, max_episode_steps=None):
    assert ('NoFrameskip' in env.spec.id)
    env = NoopResetEnv(env, noop_max=30)
    env = MaxAndSkipEnv(env, skip=4)
    assert (max_episode_steps is None)
    return env