admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-24 19:57:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(jobs): annotate ensure_dataset_available api param as HfApi

Browse Source 
Add the missing HfApi type hint via a TYPE_CHECKING import.
This commit is contained in:
Nicolas Rabault
2026-06-24 11:09:03 +02:00
parent 2d9e286f18
commit 79fd82443b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/jobs/dataset.py
+5 -1
View File
@@ -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