wrap dm_control aloha into gymnasium (TODO: properly seeding the env)

This commit is contained in:
Cadene
2024-04-05 17:17:14 +00:00
parent 26602269cd
commit 29032fbcd3
3 changed files with 104 additions and 232 deletions

View File

@@ -30,7 +30,14 @@ def make_env(cfg, num_parallel_envs=0) -> gym.Env | gym.vector.SyncVectorEnv:
**kwargs,
)
elif cfg.env.name == "aloha":
from lerobot.common.envs import aloha as gym_aloha # noqa: F401
kwargs["task"] = cfg.env.task
env_fn = lambda: gym.make( # noqa: E731
"gym_aloha/AlohaInsertion-v0",
**kwargs,
)
else:
raise ValueError(cfg.env.name)