diff --git a/tests/jobs/test_hf.py b/tests/jobs/test_hf.py index 90ebd9c67..3b275cb95 100644 --- a/tests/jobs/test_hf.py +++ b/tests/jobs/test_hf.py @@ -21,6 +21,8 @@ import draccus import httpx import pytest +pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])") + from lerobot.configs.train import TrainPipelineConfig from lerobot.jobs.hf import ( _pod_forwarded_args,