setup.py: Fix numpy installation

saketkc · saketkc · commit eab4f7c2cc39 · 2015-03-04T01:41:00.000-08:00
Pass build_requires if numpy is not found Fixes scikit-learn#4164 Signed-off-by: Saket Choudhary <saketkc@gmail.com>
diff --git a/setup.py b/setup.py
@@ -104,8 +104,19 @@ def configuration(parent_package='', top_path=None):
 
     return config
 
+def is_Numpy_installed():
+    try:
+        import numpy
+    except ImportError:
+        return False
 
 def setup_package():
+
+    build_requires = []
+    is_numpy_installed = is_Numpy_installed()
+    if is_numpy_installed is False:
+        build_requires = ['numpy>=1.6.2']
+
     metadata = dict(name=DISTNAME,
                     maintainer=MAINTAINER,
                     maintainer_email=MAINTAINER_EMAIL,
@@ -134,6 +145,7 @@ def setup_package():
                                  'Programming Language :: Python :: 3.4',
                                  ],
                     cmdclass={'clean': CleanCommand},
+                    install_requires=build_requires,
                     **extra_setuptools_args)
 
     if (len(sys.argv) >= 2
@@ -152,6 +164,10 @@ def setup_package():
 
         metadata['version'] = VERSION
     else:
+        if is_numpy_installed is False:
+            sys.stderr.write("[Error] Numpy is required to setup scikit-learn\n"
+                             "Please install numpy from http://numpy.org\n")
+            sys.exit(1)
         from numpy.distutils.core import setup
 
         metadata['configuration'] = configuration
