diff --git a/src/lerobot/datasets/video_utils.py b/src/lerobot/datasets/video_utils.py index 9e8112817..437cad76b 100644 --- a/src/lerobot/datasets/video_utils.py +++ b/src/lerobot/datasets/video_utils.py @@ -856,7 +856,7 @@ class StreamingVideoEncoder: self._closed = False def start_episode( - self, video_keys: list[str], temp_dir: Path, depth_video_keys: list[str] | None = [] + self, video_keys: list[str], temp_dir: Path, depth_video_keys: list[str] | None = None ) -> None: """Start encoder threads for a new episode. @@ -871,6 +871,9 @@ class StreamingVideoEncoder: self._dropped_frames.clear() + if depth_video_keys is None: + depth_video_keys = [] + for video_key in video_keys: frame_queue: queue.Queue = queue.Queue(maxsize=self.queue_maxsize) result_queue: queue.Queue = queue.Queue(maxsize=1)