Fixed failing tests

torchvision/transforms/v2/_geometry.py

@@ -1060,8 +1060,8 @@ def __init__(
         fill: Union[_FillType, Dict[Union[Type, str], _FillType]] = 0,
     ) -> None:
         super().__init__()
-        self.alpha = _setup_number_or_seq(alpha, "alpha", 2)
-        self.sigma = _setup_number_or_seq(sigma, "sigma", 2)
+        self.alpha = _setup_number_or_seq(alpha, "alpha")
+        self.sigma = _setup_number_or_seq(sigma, "sigma")
 
         self.interpolation = _check_interpolation(interpolation)
         self.fill = fill

torchvision/transforms/v2/_misc.py

@@ -198,7 +198,7 @@ def __init__(
             if ks <= 0 or ks % 2 == 0:
                 raise ValueError("Kernel size value should be an odd and positive number.")
 
-        self.sigma = _setup_number_or_seq(sigma, "sigma", 2)
+        self.sigma = _setup_number_or_seq(sigma, "sigma")
 
         if not 0.0 < self.sigma[0] <= self.sigma[1]:
             raise ValueError(f"sigma values should be positive and of the form (min, max). Got {self.sigma}")

torchvision/transforms/v2/_utils.py

@@ -21,7 +21,7 @@
 def _setup_number_or_seq(arg: Union[int, float, Sequence[Union[int, float]]], name: str) -> Sequence[float]:
     if not isinstance(arg, (int, float, Sequence)):
         raise TypeError(f"{name} should be a number or a sequence of numbers. Got {type(arg)}")
-    if isinstance(arg, Sequence) and len(arg) not in (1, req_size):
+    if isinstance(arg, Sequence) and len(arg) not in (1, 2):
         raise ValueError(f"If {name} is a sequence its length should be 1 or 2. Got {len(arg)}")
     if isinstance(arg, Sequence):
         for element in arg:
@@ -30,7 +30,7 @@ def _setup_number_or_seq(arg: Union[int, float, Sequence[Union[int, float]]], na
 
     if isinstance(arg, (int, float)):
         arg = [float(arg), float(arg)]
-    if isinstance(arg, Sequence):
+    elif isinstance(arg, Sequence):
         if len(arg) == 1:
             arg = [float(arg[0]), float(arg[0])]
         else: