cache in test_plotting

microsoft · Oct 29, 2020 · dfb0fd3 · dfb0fd3
1 parent 4c2e7be
commit dfb0fd3
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/tests/python_package_test/test_basic.py b/tests/python_package_test/test_basic.py
@@ -10,7 +10,7 @@
 from sklearn.datasets import dump_svmlight_file, load_svmlight_file
 from sklearn.model_selection import train_test_split
 
-from .utils import load_breast_cancer, lru_cache
+from .utils import load_breast_cancer
 
 
 class TestBasic(unittest.TestCase):

diff --git a/tests/python_package_test/test_engine.py b/tests/python_package_test/test_engine.py
@@ -19,7 +19,7 @@
 except ImportError:
     import pickle
 
-from .utils import load_boston, load_breast_cancer, load_digits, load_iris, lru_cache
+from .utils import load_boston, load_breast_cancer, load_digits, load_iris
 
 
 decreasing_generator = itertools.count(0, -1)

diff --git a/tests/python_package_test/test_plotting.py b/tests/python_package_test/test_plotting.py
@@ -3,7 +3,6 @@
 
 import lightgbm as lgb
 from lightgbm.compat import MATPLOTLIB_INSTALLED, GRAPHVIZ_INSTALLED
-from sklearn.datasets import load_breast_cancer
 from sklearn.model_selection import train_test_split
 
 if MATPLOTLIB_INSTALLED:
@@ -12,6 +11,8 @@
 if GRAPHVIZ_INSTALLED:
     import graphviz
 
+from .utils import load_breast_cancer
+
 
 class TestBasic(unittest.TestCase):
 

diff --git a/tests/python_package_test/test_sklearn.py b/tests/python_package_test/test_sklearn.py
@@ -20,7 +20,7 @@
                                             check_parameters_default_constructible)
 from sklearn.utils.validation import check_is_fitted
 
-from .utils import load_boston, load_breast_cancer, load_digits, load_iris, load_linnerud, lru_cache
+from .utils import load_boston, load_breast_cancer, load_digits, load_iris, load_linnerud
 
 
 decreasing_generator = itertools.count(0, -1)