update

CodSpeedHQ · Sep 11, 2024 · 6835aee · 6835aee
1 parent 1625fd3
commit 6835aee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/kornia/utils/sample.py b/kornia/utils/sample.py
@@ -36,10 +36,10 @@ def download_image(url: str, save_to: str) -> None:
 
 def get_sample_images(
     resize: Optional[tuple[int, int]] = None,
-    paths: list[str] = IMAGE_URLS,
+    paths: List[str] = IMAGE_URLS,
     download: bool = True,
     cache_dir: Optional[str] = None,
-) -> Union[Tensor, list[Tensor]]:
+) -> 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.

diff --git a/requirements/requirements-docs.txt b/requirements/requirements-docs.txt
@@ -1,6 +1,7 @@
 furo
 kornia_moons
 matplotlib
+onnx
 opencv-python
 PyYAML>=5.1
 sphinx