Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/huggingface_hub/cli/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def _download(self) -> str:
force_download=self.force_download,
token=self.token,
local_dir=self.local_dir,
library_name="hf",
library_name="huggingface-cli",
)

# Otherwise: use `snapshot_download` to ensure all files comes from same revision
Expand All @@ -176,6 +176,6 @@ def _download(self) -> str:
cache_dir=self.cache_dir,
token=self.token,
local_dir=self.local_dir,
library_name="hf",
library_name="huggingface-cli",
max_workers=self.max_workers,
)
2 changes: 1 addition & 1 deletion src/huggingface_hub/cli/repo_files.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def __init__(self, args) -> None:
self.repo_id: str = args.repo_id
self.repo_type: Optional[str] = args.repo_type
self.revision: Optional[str] = args.revision
self.api: HfApi = HfApi(token=args.token, library_name="hf")
self.api: HfApi = HfApi(token=args.token, library_name="huggingface-cli")
self.patterns: List[str] = args.patterns
self.commit_message: Optional[str] = args.commit_message
self.commit_description: Optional[str] = args.commit_description
Expand Down
2 changes: 1 addition & 1 deletion src/huggingface_hub/cli/upload.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def __init__(self, args: Namespace) -> None:
self.commit_message: Optional[str] = args.commit_message
self.commit_description: Optional[str] = args.commit_description
self.create_pr: bool = args.create_pr
self.api: HfApi = HfApi(token=args.token, library_name="hf")
self.api: HfApi = HfApi(token=args.token, library_name="huggingface-cli")
self.quiet: bool = args.quiet # disable warnings and progress bars

# Check `--every` is valid
Expand Down
2 changes: 1 addition & 1 deletion src/huggingface_hub/cli/upload_large_folder.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def __init__(self, args: Namespace) -> None:
self.include: Optional[List[str]] = args.include
self.exclude: Optional[List[str]] = args.exclude

self.api: HfApi = HfApi(token=args.token, library_name="hf")
self.api: HfApi = HfApi(token=args.token, library_name="huggingface-cli")

self.num_workers: Optional[int] = args.num_workers
self.no_report: bool = args.no_report
Expand Down
8 changes: 4 additions & 4 deletions tests/test_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -478,7 +478,7 @@ def test_download_file_from_revision(self, mock: Mock) -> None:
force_download=False,
token="hf_****",
local_dir=".",
library_name="hf",
library_name="huggingface-cli",
)

@patch("huggingface_hub.cli.download.snapshot_download")
Expand Down Expand Up @@ -510,7 +510,7 @@ def test_download_multiple_files(self, mock: Mock) -> None:
cache_dir=None,
token="hf_****",
local_dir="/path/to/dir",
library_name="hf",
library_name="huggingface-cli",
max_workers=8,
)

Expand Down Expand Up @@ -543,7 +543,7 @@ def test_download_with_patterns(self, mock: Mock) -> None:
cache_dir=None,
local_dir=None,
token=None,
library_name="hf",
library_name="huggingface-cli",
max_workers=8,
)

Expand Down Expand Up @@ -579,7 +579,7 @@ def test_download_with_ignored_patterns(self, mock: Mock) -> None:
cache_dir=None,
token=None,
local_dir=None,
library_name="hf",
library_name="huggingface-cli",
max_workers=8,
)

Expand Down
4 changes: 3 additions & 1 deletion tests/test_hf_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -2535,7 +2535,9 @@ def test_inference_provider_mapping_model_info(self):
assert item.provider_id is not None

def test_inference_provider_mapping_list_models(self):
models = list(self._api.list_models(author="deepseek-ai", expand="inferenceProviderMapping", limit=1))
models = list(
self._api.list_models(author="deepseek-ai", expand="inferenceProviderMapping", limit=1, inference="warm")
)
assert len(models) > 0
mapping = models[0].inference_provider_mapping
assert isinstance(mapping, list)
Expand Down
Loading