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
This commit is contained in:
Jeremy Reizenstein
2026-01-14 08:57:43 -08:00
committed by meta-codesync[bot]
parent 5b1cce56bc
commit cbcae096a0

View File

@@ -679,8 +679,8 @@ class TestTranslate(TestCaseMixin, unittest.TestCase):
im = t.inverse()._matrix im = t.inverse()._matrix
im_2 = t._matrix.inverse() im_2 = t._matrix.inverse()
im_comp = t.get_matrix().inverse() im_comp = t.get_matrix().inverse()
self.assertClose(im, im_comp) self.assertClose(im, im_comp, atol=1e-4)
self.assertClose(im, im_2) self.assertClose(im, im_2, atol=1e-4)
def test_get_item(self, batch_size=5): def test_get_item(self, batch_size=5):
device = torch.device("cuda:0") device = torch.device("cuda:0")