From 3ab08a5318e557a1c2e2d2ab51873ef96219b4f0 Mon Sep 17 00:00:00 2001 From: Caroline Pascal Date: Fri, 22 May 2026 15:13:15 +0200 Subject: [PATCH] fix(imports): fixing av import in test_depth.py --- tests/datasets/test_depth.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/datasets/test_depth.py b/tests/datasets/test_depth.py index 9a6c1f942..a50fb9495 100644 --- a/tests/datasets/test_depth.py +++ b/tests/datasets/test_depth.py @@ -11,12 +11,15 @@ and legacy marker variants) lives in ``test_dataset_metadata.py``. from pathlib import Path -import av import numpy as np import PIL.Image import pytest import torch +pytest.importorskip("av", reason="av is required (install lerobot[dataset])") + +import av + from lerobot.configs import DepthEncoderConfig from lerobot.configs.video import DEPTH_QMAX, VALID_VIDEO_CODECS from lerobot.datasets.depth_utils import dequantize_depth, quantize_depth