hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

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

Browse Source
This commit is contained in:
Cadene
2024-04-05 17:17:31 +00:00
parent 29032fbcd3
commit a2d3588fca
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lerobot/common/envs/aloha/__init__.py Normal file
View File

@@ -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"},
)