diff --git a/tests/jobs/test_dataset.py b/tests/jobs/test_dataset.py index d45ac48aa..1f8b9b836 100644 --- a/tests/jobs/test_dataset.py +++ b/tests/jobs/test_dataset.py @@ -16,6 +16,8 @@ from unittest.mock import MagicMock import pytest +pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])") + from lerobot.jobs.dataset import ensure_dataset_available