wrap dm_control aloha into gymnasium (TODO: properly seeding the env)
This commit is contained in:
15
lerobot/common/envs/aloha/__init__.py
Normal file
15
lerobot/common/envs/aloha/__init__.py
Normal 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"},
|
||||||
|
)
|
||||||
Reference in New Issue
Block a user