From 8a3a411af6eb192ac496c783cc4beeabf1a2bc28 Mon Sep 17 00:00:00 2001 From: Nicolas Rabault Date: Wed, 24 Jun 2026 11:10:30 +0200 Subject: [PATCH] chore(jobs): guard LeRobotDataset import with require_package Surface a clear "install lerobot[dataset]" error if the datasets extra is missing, instead of a raw ImportError, before pushing a local dataset. --- src/lerobot/jobs/dataset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lerobot/jobs/dataset.py b/src/lerobot/jobs/dataset.py index fcccfa3c1..cc01fd9fb 100644 --- a/src/lerobot/jobs/dataset.py +++ b/src/lerobot/jobs/dataset.py @@ -24,6 +24,7 @@ from __future__ import annotations from typing import TYPE_CHECKING from lerobot.utils.constants import HF_LEROBOT_HOME +from lerobot.utils.import_utils import require_package if TYPE_CHECKING: from huggingface_hub import HfApi @@ -49,6 +50,7 @@ def ensure_dataset_available(repo_id: str, *, api: HfApi, tags: list[str] | None print(f"[dataset] '{repo_id}' is local-only; pushing to a PRIVATE Hub repo...") # Lazy import: LeRobotDataset pulls in heavy dataset deps; defer until actually needed. + require_package("datasets", extra="dataset") from lerobot.datasets import LeRobotDataset LeRobotDataset(repo_id).push_to_hub(private=True, tags=tags)