fix: forbid bool value from JPEG's quality argument type.

Author: GdoongMathew

test/test_transforms_v2.py

@@ -6190,6 +6190,11 @@ def test_transform_invalid_quality_error(self, quality):
         with pytest.raises(ValueError, match="quality must be an integer from 1 to 100"):
             transforms.JPEG(quality=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"):
+            transforms.JPEG(quality=quality)
+
 
 class TestUtils:
     # TODO: Still need to test has_all, has_any, check_type and get_bouding_boxes

torchvision/transforms/v2/_augment.py

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