huggingface-hub 0.35.0rc0__py3-none-any.whl → 0.35.1__py3-none-any.whl

huggingface_hub/cli/repo.py

@@ -43,6 +43,10 @@ class RepoCommands(BaseHuggingfaceCLICommand):
     def register_subcommand(parser: _SubParsersAction):
         repo_parser = parser.add_parser("repo", help="Manage repos on the Hub.")
         repo_subparsers = repo_parser.add_subparsers(help="huggingface.co repos related commands")
+
+        # Show help if no subcommand is provided
+        repo_parser.set_defaults(func=lambda args: repo_parser.print_help())
+
         # CREATE
         repo_create_parser = repo_subparsers.add_parser("create", help="Create a new repo on huggingface.co")
         repo_create_parser.add_argument(
@@ -179,7 +183,9 @@ class RepoTagCommand:
 
 class RepoTagCreateCommand(RepoTagCommand):
     def run(self):
-        print(f"You are about to create tag {ANSI.bold(self.args.tag)} on {self.repo_type} {ANSI.bold(self.repo_id)}")
+        print(
+            f"You are about to create tag {ANSI.bold(str(self.args.tag))} on {self.repo_type} {ANSI.bold(self.repo_id)}"
+        )
         try:
             self.api.create_tag(
                 repo_id=self.repo_id,
@@ -192,14 +198,14 @@ class RepoTagCreateCommand(RepoTagCommand):
             print(f"{self.repo_type.capitalize()} {ANSI.bold(self.repo_id)} not found.")
             exit(1)
         except RevisionNotFoundError:
-            print(f"Revision {ANSI.bold(getattr(self.args, 'revision', None))} not found.")
+            print(f"Revision {ANSI.bold(str(getattr(self.args, 'revision', None)))} not found.")
             exit(1)
         except HfHubHTTPError as e:
             if e.response.status_code == 409:
-                print(f"Tag {ANSI.bold(self.args.tag)} already exists on {ANSI.bold(self.repo_id)}")
+                print(f"Tag {ANSI.bold(str(self.args.tag))} already exists on {ANSI.bold(self.repo_id)}")
                 exit(1)
             raise e
-        print(f"Tag {ANSI.bold(self.args.tag)} created on {ANSI.bold(self.repo_id)}")
+        print(f"Tag {ANSI.bold(str(self.args.tag))} created on {ANSI.bold(self.repo_id)}")
 
 
 class RepoTagListCommand(RepoTagCommand):

huggingface_hub/commands/delete_cache.py

@@ -22,7 +22,7 @@ Usage:
 
 NOTE:
     This command is based on `InquirerPy` to build the multiselect menu in the terminal.
-    This dependency has to be installed with `pip install huggingface_hub[cli]`. Since
+    This dependency has to be installed with `pip install "huggingface_hub[cli]"`. Since
     we want to avoid as much as possible cross-platform issues, I chose a library that
     is built on top of `python-prompt-toolkit` which seems to be a reference in terminal
     GUI (actively maintained on both Unix and Windows, 7.9k stars).
@@ -88,7 +88,7 @@ def require_inquirer_py(fn: Callable) -> Callable:
         if not _inquirer_py_available:
             raise ImportError(
                 "The `delete-cache` command requires extra dependencies to work with"
-                " the TUI.\nPlease run `pip install huggingface_hub[cli]` to install"
+                ' the TUI.\nPlease run `pip install "huggingface_hub[cli]"` to install'
                 " them.\nOtherwise, disable TUI using the `--disable-tui` flag."
             )
 

huggingface_hub/commands/scan_cache.py

@@ -77,7 +77,7 @@ class ScanCacheCommand(BaseHuggingfaceCLICommand):
             if self.verbosity >= 3:
                 print(ANSI.gray(message))
                 for warning in hf_cache_info.warnings:
-                    print(ANSI.gray(warning))
+                    print(ANSI.gray(str(warning)))
             else:
                 print(ANSI.gray(message + " Use -vvv to print details."))
 

huggingface_hub/dataclasses.py

@@ -5,6 +5,7 @@ from typing import (
     Any,
     Callable,
     Dict,
+    ForwardRef,
     List,
     Literal,
     Optional,
@@ -325,6 +326,8 @@ def type_validator(name: str, value: Any, expected_type: Any) -> None:
         validator(name, value, args)
     elif isinstance(expected_type, type):  # simple types
         _validate_simple_type(name, value, expected_type)
+    elif isinstance(expected_type, ForwardRef) or isinstance(expected_type, str):
+        return
     else:
         raise TypeError(f"Unsupported type for field '{name}': {expected_type}")
 

huggingface_hub/file_download.py

@@ -267,7 +267,7 @@ def _request_wrapper(
     """Wrapper around requests methods to follow relative redirects if `follow_relative_redirects=True` even when
     `allow_redirection=False`.
 
-    A backoff mechanism retries the HTTP call on 429, 503 and 504 errors.
+    A backoff mechanism retries the HTTP call on 5xx errors and network errors.
 
     Args:
         method (`str`):
@@ -306,7 +306,7 @@ def _request_wrapper(
         return response
 
     # Perform request and return if status_code is not in the retry list.
-    response = http_backoff(method=method, url=url, **params, retry_on_exceptions=(), retry_on_status_codes=(429,))
+    response = http_backoff(method=method, url=url, **params)
     hf_raise_for_status(response)
     return response
 
@@ -317,9 +317,6 @@ def _get_file_length_from_http_response(response: requests.Response) -> Optional
 
     This function extracts the file size from the HTTP response headers, either from the
     `Content-Range` or `Content-Length` header, if available (in that order).
-        The HTTP response object containing the headers.
-        `int` or `None`: The length of the file in bytes if the information is available,
-        otherwise `None`.
 
     Args:
         response (`requests.Response`):
@@ -329,6 +326,15 @@ def _get_file_length_from_http_response(response: requests.Response) -> Optional
         `int` or `None`: The length of the file in bytes, or None if not available.
     """
 
+    # If HTTP response contains compressed body (e.g. gzip), the `Content-Length` header will
+    # contain the length of the compressed body, not the uncompressed file size.
+    # And at the start of transmission there's no way to know the uncompressed file size for gzip,
+    # thus we return None in that case.
+    content_encoding = response.headers.get("Content-Encoding", "identity").lower()
+    if content_encoding != "identity":
+        # gzip/br/deflate/zstd etc
+        return None
+
     content_range = response.headers.get("Content-Range")
     if content_range is not None:
         return int(content_range.rsplit("/")[-1])
@@ -422,11 +428,7 @@ def http_get(
     )
 
     hf_raise_for_status(r)
-    content_length = _get_file_length_from_http_response(r)
-
-    # NOTE: 'total' is the total number of bytes to download, not the number of bytes in the file.
-    #       If the file is compressed, the number of bytes in the saved file will be higher than 'total'.
-    total = resume_size + int(content_length) if content_length is not None else None
+    total: Optional[int] = _get_file_length_from_http_response(r)
 
     if displayed_filename is None:
         displayed_filename = url