Make Atari games Parameters ---------- env_id: str name of environment skip: int frame skip max_episode_steps: int max moves for an episode
(env_id, skip=4, max_episode_steps=None)
| 221 | |
| 222 | |
| 223 | def make_atari(env_id, skip=4, max_episode_steps=None): |
| 224 | """Make Atari games |
| 225 | Parameters |
| 226 | ---------- |
| 227 | env_id: str |
| 228 | name of environment |
| 229 | skip: int |
| 230 | frame skip |
| 231 | max_episode_steps: int |
| 232 | max moves for an episode |
| 233 | """ |
| 234 | env = gym.make(env_id) |
| 235 | assert 'NoFrameskip' in env.spec.id |
| 236 | env = NoopResetEnv(env, noop_max=30) |
| 237 | env = MaxAndSkipEnv(env, skip=skip) |
| 238 | if max_episode_steps is not None: |
| 239 | env = TimeLimit(env, max_episode_steps=max_episode_steps) |
| 240 | return env |
| 241 | |
| 242 | |
| 243 | def set_seed(seed): |
no test coverage detected