def from_parallel(par_env):
if isinstance(par_env, to_parallel):
return par_env.aec_env
else:
aec_env = _parallel_env_wrapper(par_env)
ordered_env = OrderEnforcingWrapper(aec_env)
return ordered_env
def raw_env(max_frames=max_frames_default, **reward_args):
return _parallel_env_wrapper(
_parallel_env(map_size, reward_args, max_frames))
def raw_env(seed=None, max_frames=1000, **reward_args):
map_size = 80
return _parallel_env_wrapper(_parallel_env(map_size, reward_args, max_frames, seed))
def raw_env(map_size=default_map_size,
max_cycles=max_cycles_default,
minimap_mode=minimap_mode_default,
**reward_args):
return _parallel_env_wrapper(
parallel_env(map_size, max_cycles, minimap_mode, **reward_args))
def raw_env(seed=None, max_frames=300):
map_size = 45
return _parallel_env_wrapper(_parallel_env(map_size, max_frames, seed))
def BaseAtariEnv(**kwargs):
return _parallel_env_wrapper(ParallelAtariEnv(**kwargs))
def raw_env(max_frames=max_frames_default):
return _parallel_env_wrapper(_parallel_env(map_size, max_frames))
