mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-24 21:19:53 +00:00
fix(imports): fixing av import in test_depth.py
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user