only import used test case and function(#22208)

5 years ago · e18a737506
parent 6e04334cec
commit e18a737506
1 changed files with 2 additions and 1 deletions
--- 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):