fix: Wrong Type Hints in augmentation.Normalize (kornia#3003)

Fix the type hints in the `__init__` of `augmentation.Normalize`. This fixes kornia#3003.
CodSpeedHQ · Aug 28, 2024 · f285e66 · f285e66
1 parent 776b26d
commit f285e66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kornia/augmentation/_2d/intensity/normalize.py b/kornia/augmentation/_2d/intensity/normalize.py
@@ -41,8 +41,8 @@ class Normalize(IntensityAugmentationBase2D):
 
     def __init__(
         self,
-        mean: Tensor | tuple[float] | list[float] | float,
-        std: Tensor | tuple[float] | list[float] | float,
+        mean: Tensor | tuple[float, ...] | list[float] | float,
+        std: Tensor | tuple[float, ...] | list[float] | float,
         p: float = 1.0,
         keepdim: bool = False,
     ) -> None: