def _thunk():
env = make_atari(env_id)
env.seed(seed + rank)
if use_monitor:
env = Monitor(env, logger.get_dir() and os.path.join(logger.get_dir(),
str(rank)))
return wrap_deepmind(env, **wrapper_kwargs)
def create_single_atari_env(env_name, seed, use_monitor, split=''):
env = atari_wrappers.make_atari(env_name)
env.seed(seed)
if use_monitor:
env = Monitor(
env, logger.get_dir() and os.path.join(logger.get_dir(), str(seed)))
env = atari_wrappers.wrap_deepmind(env, frame_stack=True)
return env