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

2026-05-15 16:49:55 +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"},
+)