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

Add aloha as package

Browse Source
This commit is contained in:
Simon Alibert
2024-03-29 16:53:00 +01:00
parent 8fc1008809
commit f17b600198
34 changed files with 46 additions and 1158 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_envs.py
View File

@@ -5,9 +5,9 @@ from torchrl.envs.utils import check_env_specs, step_mdp
from lerobot.common.datasets.factory import make_offline_buffer
from lerobot.common.envs.factory import make_env
from lerobot.common.envs.pusht.env import PushtEnv
from lerobot.common.envs.simxarm.env import SimxarmEnv
from lerobot.common.envs.aloha.env import AlohaEnv
from lerobot.common.envs.pusht import PushtEnv
from lerobot.common.envs.xarm import SimxarmEnv
from lerobot.common.envs.aloha import AlohaEnv
from .utils import DEVICE, init_config