From f285e66f8d7506235ccb39b641a03bc897746fbe Mon Sep 17 00:00:00 2001 From: Di Fang Date: Wed, 28 Aug 2024 17:56:46 +0800 Subject: [PATCH] fix: Wrong Type Hints in augmentation.Normalize (#3003) Fix the type hints in the `__init__` of `augmentation.Normalize`. This fixes #3003. --- kornia/augmentation/_2d/intensity/normalize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: