From d3d855fbc81149bea6050c3518b041583999424f Mon Sep 17 00:00:00 2001 From: Yang Yu Date: Mon, 22 Jan 2018 12:22:27 +0800 Subject: [PATCH] Fix unittest --- python/paddle/v2/fluid/tests/test_registry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/v2/fluid/tests/test_registry.py b/python/paddle/v2/fluid/tests/test_registry.py index 4fddb9126..44e50ca55 100644 --- 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]) -- GitLab