Refine reduce_op unit test and add newline at end of file

8 years ago · 1295e5ef54
parent 8b3bf28c6b
commit 1295e5ef54
2 changed files with 2 additions and 2 deletions
--- a/python/paddle/v2/framework/tests/test_reduce_op.py
+++ b/python/paddle/v2/framework/tests/test_reduce_op.py
@ -60,7 +60,7 @@ class TestKeepDimReduce(OpTest):
    def setUp(self):
        self.op_type = "reduce_sum"
        self.inputs = {'X': np.random.random((5, 6, 10)).astype("float32")}
-        self.attrs = {'dim': -2, 'keep_dim': 1}
+        self.attrs = {'dim': -2, 'keep_dim': True}
        self.outputs = {
            'Out': self.inputs['X'].sum(axis=self.attrs['dim'], keepdims=True)
        }