admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-30 21:19:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(conflicts): rely on original camera names to sort out conflicts between two cameras

Browse Source
This commit is contained in:
CarolinePascal
2026-07-30 18:49:11 +02:00
parent a4179dd6a1
commit fae2848dd7
3 changed files with 95 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/annotations/test_camera_curation.py
+26
View File
@@ -150,6 +150,32 @@ def test_build_name_mapping_and_collision():
build_name_mapping(clash, {"observation.images.cam_0": {}, "observation.images.cam_1": {}}, cfg)
def test_build_name_mapping_disambiguates_two_wrists_from_source_names():
cfg = CameraCurationConfig(view_vocabulary=VOCAB, allow_combos=True)
existing = {"observation.images.cam_left": {}, "observation.images.cam_right": {}}
verdicts = [
CameraVerdict("observation.images.cam_left", usable=True, view_label="wrist"),
CameraVerdict("observation.images.cam_right", usable=True, view_label="wrist"),
]
mapping = build_name_mapping(verdicts, existing, cfg)
assert mapping == {
"observation.images.cam_left": "observation.images.left_wrist",
"observation.images.cam_right": "observation.images.right_wrist",
}
def test_build_name_mapping_conflict_without_hint_still_errors():
# Source keys carry no directional vocab word -> cannot disambiguate -> error.
cfg = CameraCurationConfig(view_vocabulary=VOCAB, allow_combos=True)
existing = {"observation.images.cam_0": {}, "observation.images.cam_1": {}}
verdicts = [
CameraVerdict("observation.images.cam_0", usable=True, view_label="wrist"),
CameraVerdict("observation.images.cam_1", usable=True, view_label="wrist"),
]
with pytest.raises(ValueError, match="collision"):
build_name_mapping(verdicts, existing, cfg)
def test_write_report(tmp_path):
_make_min_meta(tmp_path, "observation.images.cam_0", dtype="video")
cfg = CameraCurationConfig(repo_id="user/ds", view_vocabulary=VOCAB)