admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-25 05:29:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Caroline Pascal
2026-05-22 15:13:15 +02:00
committed by GitHub
parent 5e53e6bd2f
commit 3ab08a5318
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/datasets/test_depth.py
+4 -1
View File
@@ -11,12 +11,15 @@ and legacy marker variants) lives in ``test_dataset_metadata.py``.
from pathlib import Path from pathlib import Path
import av
import numpy as np import numpy as np
import PIL.Image import PIL.Image
import pytest import pytest
import torch import torch
pytest.importorskip("av", reason="av is required (install lerobot[dataset])")
import av
from lerobot.configs import DepthEncoderConfig from lerobot.configs import DepthEncoderConfig
from lerobot.configs.video import DEPTH_QMAX, VALID_VIDEO_CODECS from lerobot.configs.video import DEPTH_QMAX, VALID_VIDEO_CODECS
from lerobot.datasets.depth_utils import dequantize_depth, quantize_depth from lerobot.datasets.depth_utils import dequantize_depth, quantize_depth