From 4fca073db1b01322fa8a7d3e99a399530bf9f08f Mon Sep 17 00:00:00 2001 From: liqitong-a <71805392+liqitong-a@users.noreply.github.com> Date: Fri, 29 Apr 2022 14:03:23 +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 2c350b512a64a..d0cb8507ffa77 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) - if dtype=='float32': + if dtype == 'float32': self.assertTrue( np.allclose( np_corr, corr.numpy(), atol=1.e-5)) @@ -68,7 +68,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) - if dtype=='float32': + if dtype == 'float32': self.assertTrue( np.allclose( np_corr, corr.numpy(), atol=1.e-5))