From 6b64642bdb530209d3cbc76865a61db40599b3d3 Mon Sep 17 00:00:00 2001 From: Nicolas Rabault Date: Thu, 25 Jun 2026 10:01:43 +0200 Subject: [PATCH] test(jobs): skip dataset import guard in base-deps test The fast test env installs base deps only, so require_package('datasets') raised ImportError before the mocked lerobot.datasets import was reached. Monkeypatch the guard to a no-op so the unit test exercises the upload logic. --- tests/jobs/test_dataset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/jobs/test_dataset.py b/tests/jobs/test_dataset.py index 9d82caf99..417f2bb1c 100644 --- a/tests/jobs/test_dataset.py +++ b/tests/jobs/test_dataset.py @@ -50,6 +50,8 @@ def test_dataset_local_only_uploads_privately(tmp_path, monkeypatch): fake_datasets_module = MagicMock() fake_datasets_module.LeRobotDataset = mock_ds_cls monkeypatch.setitem(sys.modules, "lerobot.datasets", fake_datasets_module) + # The `datasets` extra isn't installed in the base test env; skip the import guard. + monkeypatch.setattr("lerobot.jobs.dataset.require_package", lambda *a, **k: None) assert ensure_dataset_available("user/ds", api=api, tags=["lerobot", "lelab"]) is None