admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(jobs): skip test_hf if datasets extra is missing

Browse Source 
lerobot.configs.train pulls in datasets at import time, so the module
fails to collect without lerobot[dataset]. Guard with importorskip,
matching the convention in tests/training/test_multi_gpu.py.
This commit is contained in:
Nicolas Rabault
2026-06-29 14:28:43 +02:00
parent 016fc73bd1
commit 86959e3577
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/jobs/test_hf.py
+2
View File
@@ -21,6 +21,8 @@ import draccus
import httpx import httpx
import pytest import pytest
pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
from lerobot.configs.train import TrainPipelineConfig from lerobot.configs.train import TrainPipelineConfig
from lerobot.jobs.hf import ( from lerobot.jobs.hf import (
_pod_forwarded_args, _pod_forwarded_args,