Configure environment for DeepMind-style Atari.
(env, episode_life=True, clip_rewards=True, frame_stack=False)
| 225 | |
| 226 | |
| 227 | def wrap_deepmind(env, episode_life=True, clip_rewards=True, frame_stack=False): |
| 228 | """Configure environment for DeepMind-style Atari. |
| 229 | """ |
| 230 | if episode_life: |
| 231 | env = EpisodicLifeEnv(env) |
| 232 | |
| 233 | if 'FIRE' in env.unwrapped.get_action_meanings(): |
| 234 | env = FireResetEnv(env) |
| 235 | env = WarpFrame(env) |
| 236 | |
| 237 | if clip_rewards: |
| 238 | env = ClipRewardEnv(env) |
| 239 | |
| 240 | if frame_stack: |
| 241 | env = FrameStack(env, 4) |
| 242 | |
| 243 | return env |
| 244 | |
| 245 | |
| 246 | class Monitor(gym.Wrapper): |
no test coverage detected