From 419484a914dbe094b7731356e51ce1817d56ce6f Mon Sep 17 00:00:00 2001 From: Daniel Delgado Vargas Date: Mon, 6 Apr 2026 19:03:16 +0000 Subject: [PATCH] test: relax TestResize.test_bounding_boxes_correctness expectations --- test/test_transforms_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py index 1c9bfc772b3..978269821a4 100644 --- a/test/test_transforms_v2.py +++ b/test/test_transforms_v2.py @@ -945,7 +945,7 @@ def test_bounding_boxes_correctness(self, format, size, use_max_size, fn): expected = self._reference_resize_bounding_boxes(bounding_boxes, format=format, size=size, **max_size_kwarg) self._check_output_size(bounding_boxes, actual, size=size, **max_size_kwarg) - torch.testing.assert_close(actual, expected) + torch.testing.assert_close(actual, expected, atol=1.1e-5, rtol=1e-3) def _reference_resize_keypoints(self, keypoints, *, size, max_size=None): old_height, old_width = keypoints.canvas_size