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
@@ -25,6 +25,8 @@ import pytest
import torch
import torch.nn as nn
pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
from lerobot.configs.types import FeatureType, PipelineFeatureType, PolicyFeature
from lerobot.datasets.pipeline_features import aggregate_pipeline_dataset_features
from lerobot.processor import (