only import used test case and function(#22208)

PaddlePaddle · Jan 12, 2020 · e18a737 · e18a737
1 parent 6e04334
commit e18a737
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py
@@ -15,7 +15,8 @@
 from __future__ import print_function
 
 import unittest
-from paddle.fluid.tests.unittests.test_pool2d_op import *
+import numpy as np
+from paddle.fluid.tests.unittests.test_pool2d_op import TestPool2D_Op, TestCase1, TestCase2, TestCase3, TestCase4, TestCase5, avg_pool2D_forward_naive
 
 
 def create_test_mkldnn_use_ceil_class(parent):