[fbsync] Fix Mypy (#8555)

Reviewed By: vmoens Differential Revision: D60596232 fbshipit-source-id: eeae2f7b705d64ce62bcca542d1bcf746ae36fad
pytorch · Aug 2, 2024 · 4065dac · 4065dac
1 parent a8fa77e
commit 4065dac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/torchvision/datasets/kinetics.py b/torchvision/datasets/kinetics.py
@@ -16,7 +16,7 @@
 from .vision import VisionDataset
 
 
-def _dl_wrap(tarpath: str, videopath: str, line: str) -> None:
+def _dl_wrap(tarpath: Union[str, Path], videopath:  Union[str, Path], line: str) -> None:
     download_and_extract_archive(line, tarpath, videopath)
 
 

diff --git a/torchvision/transforms/_presets.py b/torchvision/transforms/_presets.py
@@ -2,7 +2,7 @@
 This file is part of the private API. Please do not use directly these classes as they will be modified on
 future versions without warning. The classes should be accessed only via the transforms argument of Weights.
 """
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 import torch
 from torch import nn, Tensor
@@ -87,7 +87,7 @@ def __init__(
         self,
         *,
         crop_size: Tuple[int, int],
-        resize_size: Tuple[int, int],
+        resize_size: Union[Tuple[int], Tuple[int, int]],
         mean: Tuple[float, ...] = (0.43216, 0.394666, 0.37645),
         std: Tuple[float, ...] = (0.22803, 0.22145, 0.216989),
         interpolation: InterpolationMode = InterpolationMode.BILINEAR,