def make_env_func(env_config): print('Creating env!!!') cfg = default_cfg(env=spec.name) cfg.pixel_format = 'HWC' # tensorflow models expect HWC by default if 'res_w' in env_config: cfg.res_w = env_config['res_w'] if 'res_h' in env_config: cfg.res_h = env_config['res_h'] if 'renderer' in env_config: cfg.renderer = env_config['renderer'] if 'dmlab_throughput_benchmark' in env_config: cfg.renderer = env_config['dmlab_throughput_benchmark'] env = make_dmlab_env(spec.name, env_config=env_config, cfg=cfg, **kwargs) return env
def create_env(env, **kwargs): """Expected names are: doom_battle, atari_montezuma, etc.""" if env.startswith('doom_'): from envs.doom.doom_utils import make_doom_env env = make_doom_env(env, **kwargs) elif env.startswith('MiniGrid'): from envs.minigrid.minigrid_utils import make_minigrid_env env = make_minigrid_env(env, **kwargs) elif env.startswith('atari_'): from envs.atari.atari_utils import make_atari_env return make_atari_env(env, **kwargs) elif env.startswith('dmlab_'): from envs.dmlab.dmlab_env import make_dmlab_env return make_dmlab_env(env, **kwargs) elif env.startswith('quadrotor_'): from envs.quadrotors.quad_utils import make_quadrotor_env return make_quadrotor_env(env, **kwargs) else: raise Exception('Unsupported env {0}'.format(env)) return env
def make_env(env_config): from envs.dmlab.dmlab_env import make_dmlab_env return make_dmlab_env('dmlab_nonmatch', cfg=default_cfg(env='dmlab_nonmatch'), env_config=None)