Skip to content

Commit 335f840

Browse files
committed
Fix a test
1 parent da7f360 commit 335f840

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/test_transforms_v2.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5529,9 +5529,10 @@ def test_errors(self):
55295529
input_pure_tensor = input_tv_tensor.as_subclass(torch.Tensor)
55305530
format, canvas_size = input_tv_tensor.format, input_tv_tensor.canvas_size
55315531

5532-
for format_, canvas_size_ in [(None, None), (format, None), (None, canvas_size)]:
5532+
for format_, canvas_size_, clamping_mode_ in itertools.product(
5533+
(format, None), (canvas_size, None), (input_tv_tensor.clamping_mode, None)):
55335534
with pytest.raises(
5534-
ValueError, match="For pure tensor inputs, `format` and `canvas_size` have to be passed."
5535+
ValueError, match="For pure tensor inputs, `format`, `canvas_size` and `clamping_mode` have to be passed."
55355536
):
55365537
F.clamp_bounding_boxes(input_pure_tensor, format=format_, canvas_size=canvas_size_)
55375538

torchvision/transforms/v2/functional/_meta.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -551,7 +551,6 @@ def clamp_bounding_boxes(
551551

552552
if torch.jit.is_scripting() or is_pure_tensor(inpt):
553553

554-
# TODOBB
555554
if format is None or canvas_size is None or clamping_mode is None:
556555
raise ValueError("For pure tensor inputs, `format`, `canvas_size` and `clamping_mode` have to be passed.")
557556
if tv_tensors.is_rotated_bounding_format(format):

0 commit comments

Comments
 (0)