Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions test/test_transforms_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion torchvision/transforms/v2/_augment.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,))
Expand Down
Loading