From ba25a4f8dbc35429f01a1aaed88a7ed21a3628ef Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Tue, 12 May 2026 18:34:49 +0200 Subject: [PATCH] feat(codec options): passing codec options to lerobot_edit_dataset episode deletion tool --- src/lerobot/datasets/dataset_tools.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lerobot/datasets/dataset_tools.py b/src/lerobot/datasets/dataset_tools.py index 63ba47df7..3cffb4783 100644 --- a/src/lerobot/datasets/dataset_tools.py +++ b/src/lerobot/datasets/dataset_tools.py @@ -632,7 +632,8 @@ def _keep_episodes_from_video_with_av( # Convert fps to Fraction for PyAV compatibility. fps_fraction = Fraction(fps).limit_denominator(1000) - v_out = out.add_stream(camera_encoder_config.vcodec, rate=fps_fraction) + codec_options = camera_encoder_config.get_codec_options(as_strings=True) + v_out = out.add_stream(camera_encoder_config.vcodec, rate=fps_fraction, options=codec_options) # PyAV type stubs don't distinguish video streams from audio/subtitle streams. v_out.width = v_in.codec_context.width