From cbcae096a0b9b04f7c515d11bb4285a82e96b8d7 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Wed, 14 Jan 2026 08:57:43 -0800 Subject: [PATCH] Add atol=1e-4 to assertClose calls in test_inverse for Translate Summary: Added `atol=1e-4` tolerance parameter to the `assertClose` calls on lines 682 and 683 in the `test_inverse` method of `TestTranslate` class. This is a retry of D90225548 Reviewed By: sgrigory Differential Revision: D90682979 fbshipit-source-id: ac13f000174dd9962326296e1c3116d0d39c7751 --- tests/test_transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_transforms.py b/tests/test_transforms.py index 209a8a26..f2b2b2ab 100644 --- a/tests/test_transforms.py +++ b/tests/test_transforms.py @@ -679,8 +679,8 @@ class TestTranslate(TestCaseMixin, unittest.TestCase): im = t.inverse()._matrix im_2 = t._matrix.inverse() im_comp = t.get_matrix().inverse() - self.assertClose(im, im_comp) - self.assertClose(im, im_2) + self.assertClose(im, im_comp, atol=1e-4) + self.assertClose(im, im_2, atol=1e-4) def test_get_item(self, batch_size=5): device = torch.device("cuda:0")