[fbsync] allow size to be generic Sequence in Resize (#7999)

Reviewed By: vmoens Differential Revision: D50789091 fbshipit-source-id: e9d6cebe293c2c8052b2ae0f731b0a5888309829 Co-authored-by: Nicolas Hug <[email protected]> Co-authored-by: Nicolas Hug <[email protected]>
pytorch · Oct 30, 2023 · 31a0e2d · 31a0e2d
1 parent bb644e8
commit 31a0e2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py
@@ -699,7 +699,7 @@ def test_interpolation_int(self, interpolation, make_input):
         assert_equal(actual, expected)
 
     def test_transform_unknown_size_error(self):
-        with pytest.raises(ValueError, match="size can either be an integer or a list or tuple of one or two integers"):
+        with pytest.raises(ValueError, match="size can either be an integer or a sequence of one or two integers"):
             transforms.Resize(size=object())
 
     @pytest.mark.parametrize(

diff --git a/torchvision/transforms/v2/_geometry.py b/torchvision/transforms/v2/_geometry.py
@@ -135,11 +135,11 @@ def __init__(
 
         if isinstance(size, int):
             size = [size]
-        elif isinstance(size, (list, tuple)) and len(size) in {1, 2}:
+        elif isinstance(size, Sequence) and len(size) in {1, 2}:
             size = list(size)
         else:
             raise ValueError(
-                f"size can either be an integer or a list or tuple of one or two integers, " f"but got {size} instead."
+                f"size can either be an integer or a sequence of one or two integers, but got {size} instead."
             )
         self.size = size