admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +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/dataset_factories.py
Vendored
+1 -4
View File
@@ -46,7 +46,6 @@ from tests.fixtures.constants import (
DUMMY_MOTOR_FEATURES,
DUMMY_REPO_ID,
DUMMY_ROBOT_TYPE,
DUMMY_VIDEO_INFO,
)
@@ -134,9 +133,7 @@ def features_factory():
use_videos: bool = True,
) -> dict:
if use_videos:
camera_ft = {
key: {"dtype": "video", **ft, **DUMMY_VIDEO_INFO} for key, ft in camera_features.items()
}
camera_ft = {key: {"dtype": "video", **ft} for key, ft in camera_features.items()}
else:
camera_ft = {key: {"dtype": "image", **ft} for key, ft in camera_features.items()}
return {