admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-01 15:17:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(codec options): passing codec options to lerobot_edit_dataset episode deletion tool

Browse Source
This commit is contained in:
CarolinePascal
2026-05-12 18:34:49 +02:00
parent 5b976cc926
commit ba25a4f8db
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/dataset_tools.py
+2 -1
View File
@@ -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