diff --git a/src/lerobot/datasets/video_utils.py b/src/lerobot/datasets/video_utils.py index ab373412a..415841009 100644 --- a/src/lerobot/datasets/video_utils.py +++ b/src/lerobot/datasets/video_utils.py @@ -699,7 +699,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. @@ -714,6 +714,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)