From 0b379e9c4e4d748c23eab6c015d707be56a1f1c0 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Sat, 8 Mar 2025 11:12:58 +0300 Subject: [PATCH] update benchmark to new dataset format --- benchmarks/video/run_video_benchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/video/run_video_benchmark.py b/benchmarks/video/run_video_benchmark.py index e9066487..c62578c4 100644 --- a/benchmarks/video/run_video_benchmark.py +++ b/benchmarks/video/run_video_benchmark.py @@ -67,7 +67,7 @@ def parse_int_or_none(value) -> int | None: def check_datasets_formats(repo_ids: list) -> None: for repo_id in repo_ids: dataset = LeRobotDataset(repo_id) - if dataset.video: + if len(dataset.meta.video_keys) > 0: raise ValueError( f"Use only image dataset for running this benchmark. Video dataset provided: {repo_id}" )