Handle no Cython error in setup.py (#931)

Signed-off-by: Serge Panev <[email protected]>
dmlc · Sep 6, 2019 · dc1af56 · dc1af56
1 parent 384f30e
commit dc1af56
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -5,11 +5,17 @@
 import sys
 
 import numpy as np
-from Cython.Build import cythonize
+try:
+    from Cython.Build import cythonize
+except ImportError:
+    cythonize = None
 from setuptools import setup, find_packages, Extension
 
 with_cython = False
 if '--with-cython' in sys.argv:
+    if not cythonize:
+        print("Cython not found, please run `pip install Cython`")
+        exit(1)
     with_cython = True
     sys.argv.remove('--with-cython')