Update test_corr.py

PaddlePaddle · Apr 24, 2022 · decb986 · decb986
1 parent 18339f5
commit decb986
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_corr.py b/python/paddle/fluid/tests/unittests/test_corr.py
@@ -44,7 +44,7 @@ def test_tensor_corr_default(self):
                 tensor = paddle.to_tensor(np_arr, place=p)
                 corr = paddle.linalg.corrcoef(tensor)
                 np_corr = numpy_corr(np_arr, rowvar=True, dtype=dtype)
-                self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-2))
+                self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-4))
 
     def test_tensor_corr_rowvar(self):
         typelist = ['float64', 'float32']
@@ -63,7 +63,7 @@ def test_tensor_corr_rowvar(self):
                 tensor = paddle.to_tensor(np_arr, place=p)
                 corr = paddle.linalg.corrcoef(tensor, rowvar=False)
                 np_corr = numpy_corr(np_arr, rowvar=False, dtype=dtype)
-                self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-2))
+                self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-4))
 
 
 # Input(x) only support N-D (1<=N<=2) tensor