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))