defer numpy import to allow install_requires to do its job (fixes #24)

1 files changed, 9 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index f2315f2..afcfa16 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,6 @@ Distributed under the BSD License
 from setuptools import setup, find_packages
 from distutils.extension import Extension
 import os.path
-import numpy
 
 try:
     from Cython.Distutils import build_ext
@@ -21,6 +20,14 @@ except ImportError:
 else:
     use_cython = True
 
+
+class numpy_include(object):
+    """Defers import of numpy until install_requires is through"""
+    def __str__(self):
+        import numpy
+        return numpy.get_include()
+
+
 if os.path.isfile("pyspike/cython/cython_add.c") and \
    os.path.isfile("pyspike/cython/cython_profiles.c") and \
    os.path.isfile("pyspike/cython/cython_distances.c"):
@@ -58,7 +65,7 @@ setup(
     version='0.5.2',
     cmdclass=cmdclass,
     ext_modules=ext_modules,
-    include_dirs=[numpy.get_include()],
+    include_dirs=[numpy_include()],
     description='A Python library for the numerical analysis of spike\
 train similarity',
     author='Mario Mulansky',