style

Author: Wauplin

src/huggingface_hub/fastai_utils.py

@@ -6,8 +6,7 @@
 
 from packaging import version
 
-from huggingface_hub import snapshot_download
-from huggingface_hub import constants
+from huggingface_hub import constants, snapshot_download
 from huggingface_hub.hf_api import HfApi
 from huggingface_hub.utils import (
     SoftTemporaryDirectory,

src/huggingface_hub/file_download.py

@@ -18,35 +18,19 @@
 
 import requests
 
-from . import (__version__ , # noqa: F401 # for backward compatibility
-               constants,
+from . import (
+    __version__,  # noqa: F401 # for backward compatibility
+    constants,
 )
 from ._local_folder import (
     get_local_download_paths,
     read_download_metadata,
     write_download_metadata,
 )
 from .constants import (
-    DEFAULT_ETAG_TIMEOUT,
-    DEFAULT_REQUEST_TIMEOUT,
-    DEFAULT_REVISION,
-    DOWNLOAD_CHUNK_SIZE,
-    ENDPOINT,
-    HF_HUB_CACHE,
-    HF_HUB_DISABLE_SYMLINKS_WARNING,
-    HF_HUB_DOWNLOAD_TIMEOUT,
-    HF_HUB_ENABLE_HF_TRANSFER,
-    HF_HUB_ETAG_TIMEOUT,
-    HF_TRANSFER_CONCURRENCY,
     HUGGINGFACE_CO_URL_TEMPLATE,  # noqa: F401 # for backward compatibility
-    HUGGINGFACE_HEADER_X_LINKED_ETAG,
-    HUGGINGFACE_HEADER_X_LINKED_SIZE,
-    HUGGINGFACE_HEADER_X_REPO_COMMIT,
     HUGGINGFACE_HUB_CACHE,  # noqa: F401 # for backward compatibility
-    REPO_ID_SEPARATOR,
-    REPO_TYPES,
-    REPO_TYPES_URL_PREFIXES,
-)
+    )
 from .errors import (
     EntryNotFoundError,
     FileMetadataError,
@@ -1695,7 +1679,9 @@ def get_hf_file_metadata(
         # Do not use directly `url`, as `_request_wrapper` might have followed relative
         # redirects.
         location=r.headers.get("Location") or r.request.url,  # type: ignore
-        size=_int_or_none(r.headers.get(constants.HUGGINGFACE_HEADER_X_LINKED_SIZE) or r.headers.get("Content-Length")),
+        size=_int_or_none(
+            r.headers.get(constants.HUGGINGFACE_HEADER_X_LINKED_SIZE) or r.headers.get("Content-Length")
+        ),
     )
 
 

src/huggingface_hub/hf_file_system.py

@@ -15,8 +15,8 @@
 from fsspec.utils import isfilelike
 from requests import Response
 
-from ._commit_api import CommitOperationCopy, CommitOperationDelete
 from . import constants
+from ._commit_api import CommitOperationCopy, CommitOperationDelete
 from .errors import EntryNotFoundError, RepositoryNotFoundError, RevisionNotFoundError
 from .file_download import hf_hub_url, http_get
 from .hf_api import HfApi, LastCommitInfo, RepoFile
@@ -118,7 +118,9 @@ def _repo_and_revision_exist(
     ) -> Tuple[bool, Optional[Exception]]:
         if (repo_type, repo_id, revision) not in self._repo_and_revision_exists_cache:
             try:
-                self._api.repo_info(repo_id, revision=revision, repo_type=repo_type, timeout=constants.HF_HUB_ETAG_TIMEOUT)
+                self._api.repo_info(
+                    repo_id, revision=revision, repo_type=repo_type, timeout=constants.HF_HUB_ETAG_TIMEOUT
+                )
             except (RepositoryNotFoundError, HFValidationError) as e:
                 self._repo_and_revision_exists_cache[(repo_type, repo_id, revision)] = False, e
                 self._repo_and_revision_exists_cache[(repo_type, repo_id, None)] = False, e

tests/test_file_download.py

@@ -592,7 +592,9 @@ def test_try_to_load_from_cache_specific_pr_revision_exists(self):
         self.assertIsNone(try_to_load_from_cache(DUMMY_MODEL_ID, filename="conf.json", revision="refs/pr/1"))
 
         # If revision does not exist, returns None
-        self.assertIsNone(try_to_load_from_cache(DUMMY_MODEL_ID, filename=constants.CONFIG_NAME, revision="does-not-exist"))
+        self.assertIsNone(
+            try_to_load_from_cache(DUMMY_MODEL_ID, filename=constants.CONFIG_NAME, revision="does-not-exist")
+        )
 
     def test_try_to_load_from_cache_no_exist(self):
         # Make sure the file is cached
@@ -1255,7 +1257,9 @@ def test_resolve_endpoint_on_lfs_file(self):
     @staticmethod
     def _get_etag_and_normalize(response: Response) -> str:
         response.raise_for_status()
-        return _normalize_etag(response.headers.get(constants.HUGGINGFACE_HEADER_X_LINKED_ETAG) or response.headers.get("ETag"))
+        return _normalize_etag(
+            response.headers.get(constants.HUGGINGFACE_HEADER_X_LINKED_ETAG) or response.headers.get("ETag")
+        )
 
 
 @with_production_testing

tests/test_hub_mixin_pytorch.py

@@ -9,8 +9,7 @@
 
 import pytest
 
-from huggingface_hub import HfApi, ModelCard, hf_hub_download
-from huggingface_hub import constants
+from huggingface_hub import HfApi, ModelCard, constants, hf_hub_download
 from huggingface_hub.errors import EntryNotFoundError, HfHubHTTPError
 from huggingface_hub.hub_mixin import ModelHubMixin, PyTorchModelHubMixin
 from huggingface_hub.serialization._torch import storage_ptr

tests/test_lfs.py

@@ -5,7 +5,6 @@
 
 from huggingface_hub.lfs import SliceFileObj, UploadInfo
 from huggingface_hub.utils import SoftTemporaryDirectory
-from huggingface_hub import file_download
 
 
 class TestUploadInfo(unittest.TestCase):

tests/test_repocard.py

@@ -29,14 +29,14 @@
     RepoCard,
     SpaceCard,
     SpaceCardData,
+    constants,
     get_hf_file_metadata,
     hf_hub_url,
     metadata_eval_result,
     metadata_load,
     metadata_save,
     metadata_update,
 )
-from huggingface_hub import constants
 from huggingface_hub.errors import EntryNotFoundError
 from huggingface_hub.file_download import hf_hub_download
 from huggingface_hub.hf_api import HfApi
@@ -278,7 +278,9 @@ def setUp(self) -> None:
 
         self.repo_id = self.api.create_repo(repo_name()).repo_id
         self.api.upload_file(
-            path_or_fileobj=DUMMY_MODELCARD_EVAL_RESULT.encode(), repo_id=self.repo_id, path_in_repo=constants.REPOCARD_NAME
+            path_or_fileobj=DUMMY_MODELCARD_EVAL_RESULT.encode(),
+            repo_id=self.repo_id,
+            path_in_repo=constants.REPOCARD_NAME,
         )
         self.existing_metadata = yaml.safe_load(DUMMY_MODELCARD_EVAL_RESULT.strip().strip("-"))
 
@@ -413,7 +415,9 @@ def test_update_metadata_on_empty_text_content(self) -> None:
         """
         # Create modelcard with metadata but empty text content
         self.api.upload_file(
-            path_or_fileobj=DUMMY_MODELCARD_NO_TEXT_CONTENT.encode(), path_in_repo=constants.REPOCARD_NAME, repo_id=self.repo_id
+            path_or_fileobj=DUMMY_MODELCARD_NO_TEXT_CONTENT.encode(),
+            path_in_repo=constants.REPOCARD_NAME,
+            repo_id=self.repo_id,
         )
         metadata_update(self.repo_id, {"tag": "test"}, token=self.token)