fix osme missing atols

Author: ArthurZucker

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]:

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)