From 1f0ae6e298349f033a738be559bc52666e001a82 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 10 Sep 2024 19:59:08 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- docs/source/onnx.rst | 2 +- kornia/color/gray.py | 2 +- kornia/geometry/transform/affwarp.py | 2 +- kornia/utils/__init__.py | 2 +- kornia/utils/sample.py | 19 ++++++++++--------- 5 files changed, 14 insertions(+), 13 deletions(-) diff --git a/docs/source/onnx.rst b/docs/source/onnx.rst index ff3381c562..626313db2b 100644 --- a/docs/source/onnx.rst +++ b/docs/source/onnx.rst @@ -133,4 +133,4 @@ API Documentation # Load a local converted/downloaded operator onnx_loader.load_model("operators/kornia.color.gray.GrayscaleToRgb") - :members: \ No newline at end of file + :members: diff --git a/kornia/color/gray.py b/kornia/color/gray.py index c3420f60d9..f647cfd337 100644 --- a/kornia/color/gray.py +++ b/kornia/color/gray.py @@ -1,6 +1,6 @@ from __future__ import annotations -from typing import ClassVar, List, Optional +from typing import ClassVar, Optional import torch diff --git a/kornia/geometry/transform/affwarp.py b/kornia/geometry/transform/affwarp.py index ccc18c0ec6..3abb418ec4 100644 --- a/kornia/geometry/transform/affwarp.py +++ b/kornia/geometry/transform/affwarp.py @@ -1,5 +1,5 @@ import warnings -from typing import ClassVar, Optional, Tuple, Union +from typing import Optional, Tuple, Union import torch diff --git a/kornia/utils/__init__.py b/kornia/utils/__init__.py index 95cfe1016e..61806c28f8 100644 --- a/kornia/utils/__init__.py +++ b/kornia/utils/__init__.py @@ -63,5 +63,5 @@ "is_mps_tensor_safe", "dataclass_to_dict", "dict_to_dataclass", - "get_sample_images" + "get_sample_images", ] diff --git a/kornia/utils/sample.py b/kornia/utils/sample.py index ee0d2d7351..feed129b6b 100644 --- a/kornia/utils/sample.py +++ b/kornia/utils/sample.py @@ -1,14 +1,13 @@ -import os import logging -import torch -import kornia -import requests -from io import BytesIO +import os from typing import Optional, Union -from kornia.io import load_image +import requests + +import kornia from kornia.core import Tensor, stack from kornia.core.external import PILImage as Image +from kornia.io import load_image __all__ = [ "get_sample_images", @@ -36,11 +35,13 @@ def download_image(url: str, save_to: str) -> None: def get_sample_images( - resize: tuple[int, int] = None, paths: list[str] = IMAGE_URLS, download: bool = True, - cache_dir: Optional[str] = None + resize: tuple[int, int] = None, + paths: list[str] = IMAGE_URLS, + download: bool = True, + cache_dir: Optional[str] = None, ) -> Union[Tensor, list[Tensor]]: """Loads multiple images from the given URLs. - + Optionally download them, resize them if specified, and return them as a batch of tensors or a list of tensors. Args: