From 79fd82443b426e58d2b33debea0873c7e812de65 Mon Sep 17 00:00:00 2001 From: Nicolas Rabault Date: Wed, 24 Jun 2026 11:09:03 +0200 Subject: [PATCH] chore(jobs): annotate ensure_dataset_available api param as HfApi Add the missing HfApi type hint via a TYPE_CHECKING import. --- src/lerobot/jobs/dataset.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/lerobot/jobs/dataset.py b/src/lerobot/jobs/dataset.py index efd641ab4..cb124fec5 100644 --- a/src/lerobot/jobs/dataset.py +++ b/src/lerobot/jobs/dataset.py @@ -23,9 +23,13 @@ from __future__ import annotations import os from pathlib import Path +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from huggingface_hub import HfApi -def ensure_dataset_available(repo_id: str, *, api, tags: list[str] | None = None) -> None: +def ensure_dataset_available(repo_id: str, *, api: HfApi, tags: list[str] | None = None) -> None: """Ensure repo_id resolves on the Hub, pushing a local-only dataset privately first. `tags` are attached to the dataset only when we push it (an already-on-Hub