diff --git a/src/huggingface_hub/cli/_cli_utils.py b/src/huggingface_hub/cli/_cli_utils.py index de34f1973c..2f8b08df45 100644 --- a/src/huggingface_hub/cli/_cli_utils.py +++ b/src/huggingface_hub/cli/_cli_utils.py @@ -38,7 +38,7 @@ def get_hf_api(token: Optional[str] = None) -> "HfApi": # Import here to avoid circular import from huggingface_hub.hf_api import HfApi - return HfApi(token=token, library_name="hf", library_version=__version__) + return HfApi(token=token, library_name="huggingface-cli", library_version=__version__) #### TYPER UTILS diff --git a/src/huggingface_hub/cli/download.py b/src/huggingface_hub/cli/download.py index 655e44d0f7..8db6e18f07 100644 --- a/src/huggingface_hub/cli/download.py +++ b/src/huggingface_hub/cli/download.py @@ -134,7 +134,7 @@ def run_download() -> Union[str, DryRunFileInfo, list[DryRunFileInfo]]: force_download=force_download, token=token, local_dir=local_dir, - library_name="hf", + library_name="huggingface-cli", dry_run=dry_run, ) @@ -156,7 +156,7 @@ def run_download() -> Union[str, DryRunFileInfo, list[DryRunFileInfo]]: cache_dir=cache_dir, token=token, local_dir=local_dir, - library_name="hf", + library_name="huggingface-cli", max_workers=max_workers, dry_run=dry_run, ) diff --git a/tests/test_cli.py b/tests/test_cli.py index 985ef90be5..e3bec53930 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -657,7 +657,7 @@ def test_download_basic(self, runner: CliRunner) -> None: assert kwargs["cache_dir"] is None assert kwargs["local_dir"] is None assert kwargs["token"] is None - assert kwargs["library_name"] == "hf" + assert kwargs["library_name"] == "huggingface-cli" assert kwargs["max_workers"] == 8 def test_download_with_all_options(self, runner: CliRunner) -> None: @@ -707,7 +707,7 @@ def test_download_with_all_options(self, runner: CliRunner) -> None: assert kwargs["cache_dir"] == "/tmp" assert kwargs["local_dir"] == "." assert kwargs["token"] == "my-token" - assert kwargs["library_name"] == "hf" + assert kwargs["library_name"] == "huggingface-cli" assert kwargs["max_workers"] == 4 @@ -734,7 +734,7 @@ def test_download_file_from_revision(self, mock_download: Mock, mock_snapshot: M force_download=False, token=None, local_dir=None, - library_name="hf", + library_name="huggingface-cli", dry_run=False, ) mock_snapshot.assert_not_called() @@ -764,7 +764,7 @@ def test_download_multiple_files(self, mock_download: Mock, mock_snapshot: Mock) cache_dir=None, token=None, local_dir=None, - library_name="hf", + library_name="huggingface-cli", max_workers=4, dry_run=False, ) @@ -791,7 +791,7 @@ def test_download_with_patterns(self, mock_snapshot: Mock) -> None: cache_dir=None, token=None, local_dir=None, - library_name="hf", + library_name="huggingface-cli", max_workers=8, dry_run=False, )