From 1295e5ef5467a0a068179da243c20bc05e61f921 Mon Sep 17 00:00:00 2001
From: guosheng <guosheng@baidu.com>
Date: Sun, 24 Sep 2017 16:07:14 +0800
Subject: [PATCH] Refine reduce_op unit test and add newline at end of file

---
 paddle/operators/reduce_op.cu                      | 2 +-
 python/paddle/v2/framework/tests/test_reduce_op.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/operators/reduce_op.cu b/paddle/operators/reduce_op.cu
index 2dffea3a3a..595127b858 100644
--- a/paddle/operators/reduce_op.cu
+++ b/paddle/operators/reduce_op.cu
@@ -43,4 +43,4 @@ REGISTER_OP_GPU_KERNEL(
     ops::ReduceKernel<paddle::platform::GPUPlace, float, ops::MinFunctor>);
 REGISTER_OP_GPU_KERNEL(reduce_min_grad,
                        ops::ReduceGradKernel<paddle::platform::GPUPlace, float,
-                                             ops::MaxOrMinGradFunctor>);
\ No newline at end of file
+                                             ops::MaxOrMinGradFunctor>);
diff --git a/python/paddle/v2/framework/tests/test_reduce_op.py b/python/paddle/v2/framework/tests/test_reduce_op.py
index 58951f2902..70359d60cb 100644
--- 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)
         }
-- 
GitLab