wrap dm_control aloha into gymnasium (TODO: properly seeding the env)
This commit is contained in:
parent
29032fbcd3
commit
a2d3588fca
|
@ -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"},
|
||||||
|
)
|
Loading…
Reference in New Issue