From 1779f5180edfb048dad1ba406fe6cca942b47683 Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Tue, 11 Feb 2025 10:49:21 +0100 Subject: [PATCH] Fix nighlty CIs: missing atols (#35903) fix osme missing atols --- tests/models/dpt/test_modeling_dpt.py | 2 +- tests/models/zoedepth/test_modeling_zoedepth.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/dpt/test_modeling_dpt.py b/tests/models/dpt/test_modeling_dpt.py index c00b810cfe..520d87081e 100644 --- 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]: diff --git a/tests/models/zoedepth/test_modeling_zoedepth.py b/tests/models/zoedepth/test_modeling_zoedepth.py index aef49c4752..ee1da96880 100644 --- a/tests/models/zoedepth/test_modeling_zoedepth.py +++ b/tests/models/zoedepth/test_modeling_zoedepth.py @@ -323,7 +323,7 @@ class ZoeDepthModelIntegrationTest(unittest.TestCase): for img, out, expected_slice in zip(images, outputs, expected_slices): out = out["predicted_depth"] self.assertTrue(img.size == out.shape[::-1]) - torch.testing.assert_close(expected_slice, out[:3, :3], rtol=1e-3) + torch.testing.assert_close(expected_slice, out[:3, :3], atol=1e-3, rtol=1e-3) self.check_target_size(image_processor, pad_input, images, outputs, raw_outputs, raw_outputs_flipped)