fix(imports): fixing av import in test_depth.py

This commit is contained in:
Caroline Pascal
2026-05-22 15:13:15 +02:00
committed by GitHub
parent 5e53e6bd2f
commit 3ab08a5318
+4 -1
View File
@@ -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