backward compatible cli tracking (v0.x) (#3460)

* backward compatible cli tracking

* fix test

Author: Wauplin

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,
         )

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

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

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

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,
         )
 

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)