admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-16 09:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

license + peft-dep + init groot + flat import layering utils dataset

Browse Source
This commit is contained in:
Steven Palma
2026-04-12 16:43:24 +02:00
parent 718d2fc59d
commit 62135d846f
19 changed files with 190 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/utils.py
-27
View File
@@ -25,33 +25,6 @@ from lerobot.utils.import_utils import is_package_available
DEVICE = os.environ.get("LEROBOT_TEST_DEVICE", str(auto_select_torch_device()))
AVAILABLE_ROBOTS = [
"koch",
"koch_bimanual",
"aloha",
"so100",
"so101",
]
AVAILABLE_CAMERAS = [
"opencv",
"intelrealsense",
]
AVAILABLE_MOTORS = [
"dynamixel",
"feetech",
]
TEST_ROBOT_TYPES = []
for robot_type in AVAILABLE_ROBOTS:
TEST_ROBOT_TYPES += [(robot_type, True), (robot_type, False)]
TEST_CAMERA_TYPES = []
for camera_type in AVAILABLE_CAMERAS:
TEST_CAMERA_TYPES += [(camera_type, True), (camera_type, False)]
TEST_MOTOR_TYPES = []
for motor_type in AVAILABLE_MOTORS:
TEST_MOTOR_TYPES += [(motor_type, True), (motor_type, False)]
# Camera indices used for connecting physical cameras
OPENCV_CAMERA_INDEX = int(os.environ.get("LEROBOT_TEST_OPENCV_CAMERA_INDEX", 0))