Fix nighlty CIs: missing atols (#35903)

fix osme missing atols
huggingface · Feb 11, 2025 · 1779f51 · 1779f51
1 parent 1feebb5
commit 1779f51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/models/dpt/test_modeling_dpt.py b/tests/models/dpt/test_modeling_dpt.py
@@ -410,7 +410,7 @@ def test_post_processing_depth_estimation(self):
             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
@@ -323,7 +323,7 @@ def check_post_processing_test(self, image_processor, images, model, pad_input=T
         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)