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

2026-05-25 13:40:00 +00:00 · 2024-04-05 17:17:31 +00:00
parent 29032fbcd3
commit a2d3588fca
1 changed files with 15 additions and 0 deletions
@@ -0,0 +1,15 @@
 from gymnasium.envs.registration import register
 register(
    id="gym_aloha/AlohaInsertion-v0",
    entry_point="lerobot.common.envs.aloha.env:AlohaEnv",
    max_episode_steps=300,
    kwargs={"obs_type": "state", "task": "insertion"},
 )
 register(
    id="gym_aloha/AlohaTransferCube-v0",
    entry_point="lerobot.common.envs.aloha.env:AlohaEnv",
    max_episode_steps=300,
    kwargs={"obs_type": "state", "task": "transfer_cube"},
 )