fix leaking imports in minimal testing

This commit is contained in:
Steven Palma
2026-04-12 13:52:45 +02:00
parent 27292a3432
commit 2bf33ccb98
46 changed files with 256 additions and 121 deletions
+2
View File
@@ -17,6 +17,8 @@ import numpy as np
import pytest
import torch
pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
from lerobot.datasets.streaming_dataset import StreamingLeRobotDataset
from lerobot.datasets.utils import safe_shard
from lerobot.utils.constants import ACTION