Update lerobot/common/datasets/video_utils.py

Co-authored-by: Remi <re.cadene@gmail.com>
2025-03-04 13:27:24 +03:00 · 2025-03-04 13:27:24 +03:00 · c03b0db8aa
parent a8fcd3512d
commit c03b0db8aa
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/video_utils.py
+++ b/lerobot/common/datasets/video_utils.py
@ -50,7 +50,7 @@ def decode_video_frames(
    """
    if backend == "torchcodec":
        return decode_video_frames_torchcodec(video_path, timestamps, tolerance_s)
-    elif backend == "pyav":
+    elif backend in ["pyav", "video_reader"]:
        return decode_video_frames_torchvision(video_path, timestamps, tolerance_s, backend)
    else:
        raise ValueError(f"Unsupported video backend: {backend}")