tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add gym-aloha, rename simxarm -> xarm, refactor

Browse Source
This commit is contained in:
Simon Alibert
2024-04-08 16:18:53 +02:00
parent 5dff6d8339
commit 3f6dfa4916
15 changed files with 91 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/__init__.py
View File

@@ -27,7 +27,7 @@ from lerobot.__version__ import __version__ # noqa: F401
available_envs = [
"aloha",
"pusht",
"simxarm",
"xarm",
]
available_tasks_per_env = {
@@ -36,7 +36,7 @@ available_tasks_per_env = {
"sim_transfer_cube",
],
"pusht": ["pusht"],
"simxarm": ["lift"],
"xarm": ["lift"],
}
available_datasets_per_env = {
@@ -47,7 +47,7 @@ available_datasets_per_env = {
"aloha_sim_transfer_cube_scripted",
],
"pusht": ["pusht"],
"simxarm": ["xarm_lift_medium"],
"xarm": ["xarm_lift_medium"],
}
available_datasets = [dataset for env in available_envs for dataset in available_datasets_per_env[env]]