From a925d20ce4b2e73ff7d961d7408e6f6eda87af8f Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Tue, 26 May 2026 14:17:21 +0200 Subject: [PATCH] fix(typo): fixing typo --- src/lerobot/datasets/dataset_tools.py | 4 ++-- src/lerobot/datasets/dataset_writer.py | 2 +- tests/datasets/test_dataset_tools.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/lerobot/datasets/dataset_tools.py b/src/lerobot/datasets/dataset_tools.py index b553fe8ff..c9d8b464e 100644 --- a/src/lerobot/datasets/dataset_tools.py +++ b/src/lerobot/datasets/dataset_tools.py @@ -1899,11 +1899,11 @@ def convert_image_to_video_dataset( def _reencode_video_worker(args: tuple) -> Path: """Picklable worker for :func:`reencode_dataset`'s process pool.""" - video_path, camera_encoder, encoder_threads = args + video_path, video_encoder, encoder_threads = args reencode_video( input_video_path=video_path, output_video_path=video_path, - camera_encoder=camera_encoder, + video_encoder=video_encoder, encoder_threads=encoder_threads, overwrite=True, ) diff --git a/src/lerobot/datasets/dataset_writer.py b/src/lerobot/datasets/dataset_writer.py index deb60b259..ab5472aa8 100644 --- a/src/lerobot/datasets/dataset_writer.py +++ b/src/lerobot/datasets/dataset_writer.py @@ -211,7 +211,7 @@ class DatasetWriter: if frame_index == 0 and self._streaming_encoder is not None: self._streaming_encoder.start_episode( video_keys=list(self._meta.video_keys), - depth_video_keys=set(self._meta.video_keys) & set(self._meta.depth_keys), + depth_video_keys=list(self._meta.depth_keys), temp_dir=self._root, ) diff --git a/tests/datasets/test_dataset_tools.py b/tests/datasets/test_dataset_tools.py index 865649326..d36312920 100644 --- a/tests/datasets/test_dataset_tools.py +++ b/tests/datasets/test_dataset_tools.py @@ -1360,7 +1360,7 @@ def test_reencode_dataset_multi_key_multiprocessing( target_cfg = VideoEncoderConfig(vcodec="h264", g=6, crf=23, pix_fmt="yuv420p") - result = reencode_dataset(dataset, video_encoder=target_cfg, num_workers=2) + result = reencode_dataset(dataset, camera_encoder=target_cfg, num_workers=2) assert result is dataset