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)