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_available.py
View File

@@ -15,9 +15,9 @@ Note:
import pytest
import lerobot
from lerobot.common.envs.aloha.env import AlohaEnv
from lerobot.common.envs.pusht.env import PushtEnv
from lerobot.common.envs.simxarm.env import SimxarmEnv
from lerobot.common.envs.aloha import AlohaEnv
from lerobot.common.envs.pusht import PushtEnv
from lerobot.common.envs.xarm import SimxarmEnv
from lerobot.common.datasets.simxarm import SimxarmDataset
from lerobot.common.datasets.aloha import AlohaDataset

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