Skip to content

Commit aef48fd

Browse files
committed
match type hint in validate to gaussian_blur_image
1 parent a5d8013 commit aef48fd

File tree

1 file changed

+3
-3
lines changed
  • torchvision/transforms/v2/functional

1 file changed

+3
-3
lines changed

torchvision/transforms/v2/functional/_misc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import math
2-
from typing import List, Optional, Tuple, TYPE_CHECKING, Union
2+
from typing import List, Optional, Tuple, TYPE_CHECKING
33

44
import PIL.Image
55
import torch
@@ -107,8 +107,8 @@ def _get_gaussian_kernel2d(
107107

108108

109109
def _validate_kernel_size_and_sigma(
110-
kernel_size: List[int] | int,
111-
sigma: Optional[Union[List[float], float, int]] = None,
110+
kernel_size: List[int],
111+
sigma: Optional[List[float]] = None,
112112
) -> Tuple[List[int], List[float]]:
113113
# TODO: consider deprecating integers from sigma on the future
114114
if isinstance(kernel_size, int):

0 commit comments

Comments
 (0)