diff --git a/kornia/augmentation/_2d/intensity/normalize.py b/kornia/augmentation/_2d/intensity/normalize.py index ec5c4aaf77a..7f8386ddcbe 100644 --- 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: