Fix unittest

7 years ago · d3d855fbc8
parent 3008aab6ee
commit d3d855fbc8
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/fluid/tests/test_registry.py
+++ b/python/paddle/v2/fluid/tests/test_registry.py
@ -28,4 +28,4 @@ class TestRegistry(unittest.TestCase):
        exe = fluid.Executor(place)
        X = np.random.random((10, 10)).astype("float32")
        mean_out = exe.run(feed={"X": X}, fetch_list=[output])
-        self.assertAlmostEqual(np.mean(X), mean_out)
+        self.assertAlmostEqual(np.mean(X), mean_out[0])