diff --git a/lerobot/common/datasets/video_utils.py b/lerobot/common/datasets/video_utils.py index 9e042d92..cb81a25f 100644 --- 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}")