From c03b0db8aac7234c1f928dd2384de46291d44366 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Tue, 4 Mar 2025 13:27:24 +0300 Subject: [PATCH] Update lerobot/common/datasets/video_utils.py Co-authored-by: Remi --- lerobot/common/datasets/video_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}")