Fix nighlty CIs: missing atols (#35903)

fix osme missing atols
2025-02-11 10:49:21 +01:00
parent 1feebb5b41
commit 1779f5180e
2 changed files with 2 additions and 2 deletions
--- a/tests/models/dpt/test_modeling_dpt.py
+++ b/tests/models/dpt/test_modeling_dpt.py
@@ -410,7 +410,7 @@ class DPTModelIntegrationTest(unittest.TestCase):
            predicted_depth.unsqueeze(0).unsqueeze(1), size=(500, 500), mode="bicubic", align_corners=False
        ).squeeze()
        self.assertTrue(output_enlarged.shape == expected_shape)
-        torch.testing.assert_close(predicted_depth_l, output_enlarged, rtol=1e-3)
+        torch.testing.assert_close(predicted_depth_l, output_enlarged, atol=1e-3, rtol=1e-3)

    def test_export(self):
        for strict in [True, False]: