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

test(factories): updating factories

Browse Source
This commit is contained in:
CarolinePascal
2026-05-12 17:23:18 +02:00
parent e22ecbbcda
commit 63bbf092a0
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/fixtures/constants.py
Vendored
+10 -4
View File
@@ -28,17 +28,23 @@ DUMMY_MOTOR_FEATURES = {
"names": ["shoulder_pan", "shoulder_lift", "elbow_flex", "wrist_flex", "wrist_roll", "gripper"], "names": ["shoulder_pan", "shoulder_lift", "elbow_flex", "wrist_flex", "wrist_roll", "gripper"],
}, },
} }
DUMMY_CAMERA_FEATURES = {
"laptop": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": None},
"phone": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": None},
}
DEFAULT_FPS = 30 DEFAULT_FPS = 30
DUMMY_VIDEO_INFO = { DUMMY_VIDEO_INFO = {
"video.fps": DEFAULT_FPS, "video.fps": DEFAULT_FPS,
"video.codec": "av1", "video.codec": "av1",
"video.pix_fmt": "yuv420p", "video.pix_fmt": "yuv420p",
"video.video_backend": "pyav",
"video.extra_options": {},
"video.g": 2,
"video.crf": 30,
"video.preset": 12,
"video.fast_decode": 0,
"video.is_depth_map": False, "video.is_depth_map": False,
"has_audio": False, "has_audio": False,
} }
DUMMY_CAMERA_FEATURES = {
"laptop": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": DUMMY_VIDEO_INFO},
"phone": {"shape": (64, 96, 3), "names": ["height", "width", "channels"], "info": DUMMY_VIDEO_INFO},
}
DUMMY_CHW = (3, 96, 128) DUMMY_CHW = (3, 96, 128)
DUMMY_HWC = (96, 128, 3) DUMMY_HWC = (96, 128, 3)
tests/fixtures/dataset_factories.py
Vendored
+1 -4
View File
@@ -46,7 +46,6 @@ from tests.fixtures.constants import (
DUMMY_MOTOR_FEATURES, DUMMY_MOTOR_FEATURES,
DUMMY_REPO_ID, DUMMY_REPO_ID,
DUMMY_ROBOT_TYPE, DUMMY_ROBOT_TYPE,
DUMMY_VIDEO_INFO,
) )
@@ -134,9 +133,7 @@ def features_factory():
use_videos: bool = True, use_videos: bool = True,
) -> dict: ) -> dict:
if use_videos: if use_videos:
camera_ft = { camera_ft = {key: {"dtype": "video", **ft} for key, ft in camera_features.items()}
key: {"dtype": "video", **ft, **DUMMY_VIDEO_INFO} for key, ft in camera_features.items()
}
else: else:
camera_ft = {key: {"dtype": "image", **ft} for key, ft in camera_features.items()} camera_ft = {key: {"dtype": "image", **ft} for key, ft in camera_features.items()}
return { return {