Update test_corr.py

PaddlePaddle · Apr 29, 2022 · 4fca073 · 4fca073
1 parent e97d91c
commit 4fca073
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)
-                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))