MNT skip preprocessing.rst when pandas is not installed (scikit-learn#19016)

glemaitre · web-flow · commit 51bd34378f9c · 2020-12-18T11:17:22.000+01:00
diff --git a/doc/conftest.py b/doc/conftest.py
@@ -3,8 +3,6 @@
 from os.path import join
 import warnings
 
-import numpy as np
-
 from sklearn.utils import IS_PYPY
 from sklearn.utils._testing import SkipTest
 from sklearn.utils._testing import check_skip_network
@@ -72,6 +70,13 @@ def setup_grid_search():
         raise SkipTest("Skipping grid_search.rst, pandas not installed")
 
 
+def setup_preprocessing():
+    try:
+        import pandas  # noqa
+    except ImportError:
+        raise SkipTest("Skipping preprocessing.rst, pandas not installed")
+
+
 def setup_unsupervised_learning():
     try:
         import skimage  # noqa
@@ -105,5 +110,7 @@ def pytest_runtest_setup(item):
         setup_impute()
     elif fname.endswith('modules/grid_search.rst'):
         setup_grid_search()
+    elif fname.endswith('modules/preprocessing.rst'):
+        setup_preprocessing()
     elif fname.endswith('statistical_inference/unsupervised_learning.rst'):
         setup_unsupervised_learning()
