fix leaking imports in minimal testing

2026-05-24 04:59:47 +00:00 · 2026-04-12 13:52:45 +02:00
parent 27292a3432
commit 2bf33ccb98
46 changed files with 256 additions and 121 deletions
@@ -20,6 +20,9 @@ from pathlib import Path
 import einops
 import pytest
 import torch
+
+pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
+
 from packaging import version
 from safetensors.torch import load_file