fix test

AntoineSimoulin · AntoineSimoulin · commit ff81715d6bfb · 2025-08-24T20:53:22.000-07:00
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -7033,9 +7033,11 @@ def test_parallelogram_to_bounding_boxes(input_size, device):
         dtype=torch.float32,
     )
     expected = torch.tensor(
-        [[0, 4, 4, 0, 5, 1, 1, 5], [0, 1, 1, 0, 5, 4, 3, 4]],
+        [[0, 4, 4, 0, 5, 1, 1, 5], [0, 1, 1, 0, 5, 4, 4, 5]],
         dtype=torch.float32,
     )
+    actual = _parallelogram_to_bounding_boxes(parallelogram)
+    torch.testing.assert_close(actual, expected)
 
 
 @pytest.mark.parametrize("image_type", (PIL.Image, torch.Tensor, tv_tensors.Image))

Original file line number	Diff line number	Diff line change
`@@ -7033,9 +7033,11 @@ def test_parallelogram_to_bounding_boxes(input_size, device):`
`7033`	`7033`	`dtype=torch.float32,`
`7034`	`7034`	`)`
`7035`	`7035`	`expected = torch.tensor(`
`7036`		`- [[0, 4, 4, 0, 5, 1, 1, 5], [0, 1, 1, 0, 5, 4, 3, 4]],`
	`7036`	`+ [[0, 4, 4, 0, 5, 1, 1, 5], [0, 1, 1, 0, 5, 4, 4, 5]],`
`7037`	`7037`	`dtype=torch.float32,`
`7038`	`7038`	`)`
	`7039`	`+ actual = _parallelogram_to_bounding_boxes(parallelogram)`
	`7040`	`+ torch.testing.assert_close(actual, expected)`
`7039`	`7041`
`7040`	`7042`
`7041`	`7043`	`@pytest.mark.parametrize("image_type", (PIL.Image, torch.Tensor, tv_tensors.Image))`