mypy

Author: NicolasHug

test/test_transforms_v2.py

@@ -6208,7 +6208,7 @@ def test_transform_invalid_quality_error(self, quality):
 
     @pytest.mark.parametrize("quality", [None, True])
     def test_transform_quality_type_error(self, quality):
-        with pytest.raises(TypeError, match="quality should be a sequence of length 2"):
+        with pytest.raises(TypeError, match="quality"):
             transforms.JPEG(quality=quality)
 
 

torchvision/transforms/v2/_augment.py

@@ -351,7 +351,9 @@ class JPEG(Transform):
 
     def __init__(self, quality: Union[int, Sequence[int]]):
         super().__init__()
-        if isinstance(quality, int) and not isinstance(quality, bool):
+        if isinstance(quality, int):
+            if isinstance(quality, bool):
+                raise TypeError("quality can't be bool")
             quality = [quality, quality]
         else:
             _check_sequence_input(quality, "quality", req_sizes=(2,))