In current train.py make_env function, it seems that metadrive env is not supportive。
How can I run experiments on metaDrive?
def make_env(config, **overrides): suite, task = config.task.split('_', 1) ctor = { 'dummy': 'embodied.envs.dummy:Dummy', 'gym': 'embodied.envs.from_gym:FromGym', 'safetygym': 'embodied.envs.safetygym:SafetyGym', 'safetygymcoor': 'embodied.envs.safetygymcoor:SafetyGymCoor', 'safetygymmujoco': 'embodied.envs.safetygym_mujoco:SafetyGymMujoco', 'dm': 'embodied.envs.from_dmenv:FromDM', 'crafter': 'embodied.envs.crafter:Crafter', 'dmc': 'embodied.envs.dmc:DMC', 'atari': 'embodied.envs.atari:Atari', 'dmlab': 'embodied.envs.dmlab:DMLab', 'minecraft': 'embodied.envs.minecraft:Minecraft', 'loconav': 'embodied.envs.loconav:LocoNav', 'pinpad': 'embodied.envs.pinpad:PinPad', }[suite] if isinstance(ctor, str): module, cls = ctor.split(':') module = importlib.import_module(module) ctor = getattr(module, cls) kwargs = config.env.get(suite, {}) kwargs.update(overrides) kwargs.update({'platform':config.jax.platform}) env = ctor(task, **kwargs) return wrap_env(env, config)
In current train.py make_env function, it seems that metadrive env is not supportive。
How can I run experiments on metaDrive?
def make_env(config, **overrides): suite, task = config.task.split('_', 1) ctor = { 'dummy': 'embodied.envs.dummy:Dummy', 'gym': 'embodied.envs.from_gym:FromGym', 'safetygym': 'embodied.envs.safetygym:SafetyGym', 'safetygymcoor': 'embodied.envs.safetygymcoor:SafetyGymCoor', 'safetygymmujoco': 'embodied.envs.safetygym_mujoco:SafetyGymMujoco', 'dm': 'embodied.envs.from_dmenv:FromDM', 'crafter': 'embodied.envs.crafter:Crafter', 'dmc': 'embodied.envs.dmc:DMC', 'atari': 'embodied.envs.atari:Atari', 'dmlab': 'embodied.envs.dmlab:DMLab', 'minecraft': 'embodied.envs.minecraft:Minecraft', 'loconav': 'embodied.envs.loconav:LocoNav', 'pinpad': 'embodied.envs.pinpad:PinPad', }[suite] if isinstance(ctor, str): module, cls = ctor.split(':') module = importlib.import_module(module) ctor = getattr(module, cls) kwargs = config.env.get(suite, {}) kwargs.update(overrides) kwargs.update({'platform':config.jax.platform}) env = ctor(task, **kwargs) return wrap_env(env, config)