diff --git a/src/huggingface_hub/cli/download.py b/src/huggingface_hub/cli/download.py index 3e59233da1..2660644e62 100644 --- a/src/huggingface_hub/cli/download.py +++ b/src/huggingface_hub/cli/download.py @@ -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 @@ -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, ) diff --git a/src/huggingface_hub/cli/repo_files.py b/src/huggingface_hub/cli/repo_files.py index 34fbeb09c2..403d3126e2 100644 --- a/src/huggingface_hub/cli/repo_files.py +++ b/src/huggingface_hub/cli/repo_files.py @@ -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 diff --git a/src/huggingface_hub/cli/upload.py b/src/huggingface_hub/cli/upload.py index 07ab79bf24..0306bf9f57 100644 --- a/src/huggingface_hub/cli/upload.py +++ b/src/huggingface_hub/cli/upload.py @@ -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 diff --git a/src/huggingface_hub/cli/upload_large_folder.py b/src/huggingface_hub/cli/upload_large_folder.py index 618cd21b52..675c9ffe3d 100644 --- a/src/huggingface_hub/cli/upload_large_folder.py +++ b/src/huggingface_hub/cli/upload_large_folder.py @@ -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 diff --git a/tests/test_cli.py b/tests/test_cli.py index ae9ebc7886..bec3652ebf 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -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") @@ -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, ) @@ -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, ) @@ -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, ) diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py index 8ba57d264c..41fb24de10 100644 --- a/tests/test_hf_api.py +++ b/tests/test_hf_api.py @@ -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)