From 25e21507844372da18543a277faee2a9d4b2b724 Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Wed, 22 Nov 2017 15:16:37 -0500 Subject: [PATCH] handle conda build --- setup.py | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) diff --git a/setup.py b/setup.py index e626c66576..d3e843179c 100644 --- a/setup.py +++ b/setup.py @@ -33,18 +33,24 @@ import versioneer -try: - import Cython -except ImportError: - raise Exception("Install Cython before zipline.") -try: - import numpy as np -except ImportError: - raise Exception("Install numpy before zipline.") +conda_build = os.path.basename(sys.argv[0]) in ('conda-build', # unix + 'conda-build-script.py') # win + +if conda_build: + NumpyExtension = Extension +else: + try: + import Cython + except ImportError: + raise Exception("Install Cython before zipline.") + try: + import numpy as np + except ImportError: + raise Exception("Install numpy before zipline.") -NumpyExtension = partial(Extension, include_dirs=[np.get_include()]) + NumpyExtension = partial(Extension, include_dirs=[np.get_include()]) def window_specialization(typename): @@ -240,9 +246,6 @@ def setup_requirements(requirements_path, module_names, strict_bounds, return module_lines -conda_build = os.path.basename(sys.argv[0]) in ('conda-build', # unix - 'conda-build-script.py') # win - setup_requires = setup_requirements( 'etc/requirements.txt', ('Cython', 'numpy'),