From 7c832fa2a7f6d448943b45f3b9567a7da8918b29 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Mon, 7 Apr 2025 19:14:43 +0200 Subject: [PATCH] Taking into account situation where visual data is stored as images --- lerobot/common/datasets/lerobot_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py index 1ca61520..1966b8e7 100644 --- a/lerobot/common/datasets/lerobot_dataset.py +++ b/lerobot/common/datasets/lerobot_dataset.py @@ -210,7 +210,7 @@ class LeRobotDatasetMetadata: @property def audio_camera_keys_mapping(self) -> dict[str, str]: """Mapping between camera keys and audio keys when both are linked.""" - return {self.features[camera_key]["audio"]:camera_key for camera_key in self.camera_keys if self.features[camera_key]["audio"] is not None} + return {self.features[camera_key]["audio"]:camera_key for camera_key in self.camera_keys if self.features[camera_key]["audio"] is not None and self.features[camera_key]["dtype"] == "video"} @property def names(self) -> dict[str, list | dict]: