update ruff (#2071)

setup.py

@@ -88,7 +88,7 @@ def get_version() -> str:
 ]
 
 extras["quality"] = [
-    "ruff>=0.1.3",
+    "ruff>=0.3.0",
     "mypy==1.5.1",
 ]
 

src/huggingface_hub/_commit_api.py

@@ -1,6 +1,7 @@
 """
 Type definitions and utilities for the `create_commit` API
 """
+
 import base64
 import io
 import os

src/huggingface_hub/_login.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains methods to login to the Hub."""
+
 import os
 import subprocess
 from functools import partial

src/huggingface_hub/_multi_commits.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to multi-commits (i.e. push changes iteratively on a PR)."""
+
 import re
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Iterable, List, Optional, Set, Tuple, Union

src/huggingface_hub/_tensorboard_logger.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains a logger to push training logs to the Hub, using Tensorboard."""
+
 from pathlib import Path
 from typing import TYPE_CHECKING, List, Optional, Union
 

src/huggingface_hub/_webhooks_payload.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains data structures to parse the webhooks payload."""
+
 from typing import List, Literal, Optional
 
 from pydantic import BaseModel

src/huggingface_hub/_webhooks_server.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains `WebhooksServer` and `webhook_endpoint` to create a webhook server easily."""
+
 import atexit
 import inspect
 import os

src/huggingface_hub/commands/_cli_utils.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains a utility for good-looking prints."""
+
 import os
 from typing import List, Union
 

src/huggingface_hub/commands/delete_cache.py

@@ -55,6 +55,7 @@
 TODO: add "-y" arg for immediate deletion ?
 See discussions in https://github.com/huggingface/huggingface_hub/issues/1025.
 """
+
 import os
 from argparse import Namespace, _SubParsersAction
 from functools import wraps

src/huggingface_hub/commands/download.py

@@ -35,6 +35,7 @@
     # Download to local dir
     huggingface-cli download gpt2 --local-dir=./models/gpt2
 """
+
 import warnings
 from argparse import Namespace, _SubParsersAction
 from typing import List, Literal, Optional, Union

src/huggingface_hub/commands/env.py

@@ -16,6 +16,7 @@
 Usage:
     huggingface-cli env
 """
+
 from argparse import _SubParsersAction
 
 from ..utils import dump_environment_info

src/huggingface_hub/commands/scan_cache.py

@@ -20,6 +20,7 @@
     huggingface-cli scan-cache -vvv
     huggingface-cli scan-cache --dir ~/.cache/huggingface/hub
 """
+
 import time
 from argparse import Namespace, _SubParsersAction
 from typing import Optional

src/huggingface_hub/commands/upload.py

@@ -42,6 +42,7 @@
     # Schedule commits every 30 minutes
     huggingface-cli upload Wauplin/my-cool-model --every=30
 """
+
 import os
 import time
 import warnings

src/huggingface_hub/community.py

@@ -4,6 +4,7 @@
 See [the Discussions and Pull Requests guide](https://huggingface.co/docs/hub/repositories-pull-requests-discussions)
 for more information on Pull Requests, Discussions, and the community tab.
 """
+
 from dataclasses import dataclass
 from datetime import datetime
 from typing import List, Literal, Optional, Union

src/huggingface_hub/hf_api.py

@@ -3556,8 +3556,7 @@ def create_commit(  # type: ignore
         num_threads: int = 5,
         parent_commit: Optional[str] = None,
         run_as_future: Literal[False] = ...,
-    ) -> CommitInfo:
-        ...
+    ) -> CommitInfo: ...
 
     @overload
     def create_commit(
@@ -3574,8 +3573,7 @@ def create_commit(
         num_threads: int = 5,
         parent_commit: Optional[str] = None,
         run_as_future: Literal[True] = ...,
-    ) -> Future[CommitInfo]:
-        ...
+    ) -> Future[CommitInfo]: ...
 
     @validate_hf_hub_args
     @future_compatible
@@ -4291,8 +4289,7 @@ def upload_file(  # type: ignore
         create_pr: Optional[bool] = None,
         parent_commit: Optional[str] = None,
         run_as_future: Literal[False] = ...,
-    ) -> CommitInfo:
-        ...
+    ) -> CommitInfo: ...
 
     @overload
     def upload_file(
@@ -4309,8 +4306,7 @@ def upload_file(
         create_pr: Optional[bool] = None,
         parent_commit: Optional[str] = None,
         run_as_future: Literal[True] = ...,
-    ) -> Future[CommitInfo]:
-        ...
+    ) -> Future[CommitInfo]: ...
 
     @validate_hf_hub_args
     @future_compatible
@@ -4498,8 +4494,7 @@ def upload_folder(  # type: ignore
         multi_commits: Literal[False] = ...,
         multi_commits_verbose: bool = False,
         run_as_future: Literal[False] = ...,
-    ) -> CommitInfo:
-        ...
+    ) -> CommitInfo: ...
 
     @overload
     def upload_folder(  # type: ignore
@@ -4544,8 +4539,7 @@ def upload_folder(  # type: ignore
         multi_commits: Literal[False] = ...,
         multi_commits_verbose: bool = False,
         run_as_future: Literal[True] = ...,
-    ) -> Future[CommitInfo]:
-        ...
+    ) -> Future[CommitInfo]: ...
 
     @overload
     def upload_folder(

src/huggingface_hub/inference/_client.py

@@ -1207,8 +1207,7 @@ def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> str:
-        ...
+    ) -> str: ...
 
     @overload
     def text_generation(  # type: ignore
@@ -1231,8 +1230,7 @@ def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> TextGenerationResponse:
-        ...
+    ) -> TextGenerationResponse: ...
 
     @overload
     def text_generation(  # type: ignore
@@ -1255,8 +1253,7 @@ def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> Iterable[str]:
-        ...
+    ) -> Iterable[str]: ...
 
     @overload
     def text_generation(
@@ -1279,8 +1276,7 @@ def text_generation(
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> Iterable[TextGenerationStreamResponse]:
-        ...
+    ) -> Iterable[TextGenerationStreamResponse]: ...
 
     def text_generation(
         self,

src/huggingface_hub/inference/_common.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities used by both the sync and async inference clients."""
+
 import base64
 import io
 import json
@@ -163,13 +164,15 @@ def _first_or_none(items: List[Any]) -> Optional[Any]:
 
 
 @overload
-def _open_as_binary(content: ContentT) -> ContextManager[BinaryT]:
-    ...  # means "if input is not None, output is not None"
+def _open_as_binary(
+    content: ContentT,
+) -> ContextManager[BinaryT]: ...  # means "if input is not None, output is not None"
 
 
 @overload
-def _open_as_binary(content: Literal[None]) -> ContextManager[Literal[None]]:
-    ...  # means "if input is None, output is None"
+def _open_as_binary(
+    content: Literal[None],
+) -> ContextManager[Literal[None]]: ...  # means "if input is None, output is None"
 
 
 @contextmanager  # type: ignore

src/huggingface_hub/inference/_generated/_async_client.py

@@ -1230,8 +1230,7 @@ async def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> str:
-        ...
+    ) -> str: ...
 
     @overload
     async def text_generation(  # type: ignore
@@ -1254,8 +1253,7 @@ async def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> TextGenerationResponse:
-        ...
+    ) -> TextGenerationResponse: ...
 
     @overload
     async def text_generation(  # type: ignore
@@ -1278,8 +1276,7 @@ async def text_generation(  # type: ignore
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> AsyncIterable[str]:
-        ...
+    ) -> AsyncIterable[str]: ...
 
     @overload
     async def text_generation(
@@ -1302,8 +1299,7 @@ async def text_generation(
         truncate: Optional[int] = None,
         typical_p: Optional[float] = None,
         watermark: bool = False,
-    ) -> AsyncIterable[TextGenerationStreamResponse]:
-        ...
+    ) -> AsyncIterable[TextGenerationStreamResponse]: ...
 
     async def text_generation(
         self,

src/huggingface_hub/inference/_generated/types/base.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains a base class for all inference types."""
+
 import inspect
 import json
 import warnings

src/huggingface_hub/lfs.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Git LFS related type definitions and utilities"""
+
 import inspect
 import io
 import os

src/huggingface_hub/serialization/__init__.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # ruff: noqa: F401
 """Contains helpers to serialize tensors."""
+
 from ._base import StateDictSplit, split_state_dict_into_shards_factory
 from ._numpy import split_numpy_state_dict_into_shards
 from ._tensorflow import split_tf_state_dict_into_shards

src/huggingface_hub/serialization/_base.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains helpers to split tensors into shards."""
+
 from dataclasses import dataclass, field
 from typing import Any, Callable, Dict, List, Optional, TypeVar
 

src/huggingface_hub/serialization/_numpy.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains numpy-specific helpers."""
+
 from typing import TYPE_CHECKING, Dict
 
 from ._base import FILENAME_PATTERN, MAX_SHARD_SIZE, StateDictSplit, split_state_dict_into_shards_factory

src/huggingface_hub/serialization/_tensorflow.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains tensorflow-specific helpers."""
+
 import math
 import re
 from typing import TYPE_CHECKING, Dict

src/huggingface_hub/serialization/_torch.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains pytorch-specific helpers."""
+
 import importlib
 from functools import lru_cache
 from typing import TYPE_CHECKING, Dict, Tuple

src/huggingface_hub/utils/_cache_manager.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to manage the HF cache directory."""
+
 import os
 import shutil
 import time

src/huggingface_hub/utils/_chunk_utils.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains a utility to iterate by chunks over an iterator."""
+
 import itertools
 from typing import Iterable, TypeVar
 

src/huggingface_hub/utils/_datetime.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to handle datetimes in Huggingface Hub."""
+
 from datetime import datetime, timezone
 
 

src/huggingface_hub/utils/_experimental.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to flag a feature as "experimental" in Huggingface Hub."""
+
 import warnings
 from functools import wraps
 from typing import Callable

src/huggingface_hub/utils/_git_credential.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to manage Git credentials."""
+
 import re
 import subprocess
 from typing import List, Optional

src/huggingface_hub/utils/_headers.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contains utilities to handle headers to send in calls to Huggingface Hub."""
+
 from typing import Dict, Optional, Union
 
 from .. import constants

src/huggingface_hub/utils/_hf_folder.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Contain helper class to retrieve/store token from/to local cache."""
+
 import warnings
 from pathlib import Path
 from typing import Optional