From decb986d5308b73c5b680abf33f29ea396b14a0a Mon Sep 17 00:00:00 2001 From: liqitong-a <71805392+liqitong-a@users.noreply.github.com> Date: Sun, 24 Apr 2022 16:09:21 +0800 Subject: [PATCH] Update test_corr.py --- python/paddle/fluid/tests/unittests/test_corr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_corr.py b/python/paddle/fluid/tests/unittests/test_corr.py index 85c2801b97bab..25f17fc75d538 100644 --- 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