diff --git a/paddle/framework/shape_inference.h b/paddle/framework/shape_inference.h
index 8189823c1943e73308f18ee3ca58676b3c4ab636..bc8af0eb3ec7e8685eb7d2734e9b8f75372d1309 100644
--- a/paddle/framework/shape_inference.h
+++ b/paddle/framework/shape_inference.h
@@ -14,16 +14,11 @@ limitations under the License. */
 
 #pragma once
 
-#include "paddle/framework/attribute.h"
 #include "paddle/framework/ddim.h"
 
 namespace paddle {
 namespace framework {
 
-class InferShapeContextBase;
-
-typedef std::function<void(InferShapeContextBase *)> InferShapeFn;
-
 class InferShapeContextBase {
  public:
   virtual ~InferShapeContextBase() {}
diff --git a/python/paddle/v2/framework/tests/test_infer_shape.py b/python/paddle/v2/framework/tests/test_infer_shape.py
index ec93aaf84370d461254f68c1b352d991680de698..b38ec9c03740a2e69f1247c094ce56ab43fa8e32 100644
--- a/python/paddle/v2/framework/tests/test_infer_shape.py
+++ b/python/paddle/v2/framework/tests/test_infer_shape.py
@@ -26,8 +26,6 @@ class TestInferShape(unittest.TestCase):
         sum_op_desc.set_input("X", ["x1", "x2"])
         sum_op_desc.set_output("Out", ["out"])
 
-        print(type(sum_op_desc))
-        print(type(block))
         core.Operator.infer_shape(sum_op_desc, block)
         self.assertEqual(out.shape(), shape)