Skip to content

Commit df421b4

Browse files
zy1gitNicolasHug
andauthored
fix the error message for function _parse_pad_padding (#9334)
Co-authored-by: Nicolas Hug <[email protected]>
1 parent ec4f95a commit df421b4

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

torchvision/transforms/v2/functional/_geometry.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1566,9 +1566,7 @@ def _parse_pad_padding(padding: Union[int, list[int]]) -> list[int]:
15661566
pad_right = padding[2]
15671567
pad_bottom = padding[3]
15681568
else:
1569-
raise ValueError(
1570-
f"Padding must be an int or a 1, 2, or 4 element tuple, not a {len(padding)} element tuple"
1571-
)
1569+
raise ValueError(f"Padding must be an int or a 1, 2, or 4 element tuple or list. Found {len(padding)}")
15721570
else:
15731571
raise TypeError(f"`padding` should be an integer or tuple or list of integers, but got {padding}")
15741572

0 commit comments

Comments
 (0)