apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-27 14:25:16 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow setup.py to be parsed without numpy

Browse source 
If numpy isn't available, don't define ext_mods, pip will then determine numpy is required, install it, then call us again.

Fixes #653
This commit is contained in:
David Sheldon 2018-07-24 15:43:29 +01:00 • committed by GitHub
parent 06441f583f
commit 2ca138ea68
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

30
setup.py
View file

@ -38,7 +38,6 @@ from __future__ import print_function
import os import os
import sys import sys
from setuptools import setup, Extension from setuptools import setup, Extension
import numpy as np
import codecs import codecs
def read(fname): def read(fname):
@ -80,32 +79,39 @@ else:
compile_flags = [ '-fopenmp', '-O3'] compile_flags = [ '-fopenmp', '-O3']
link_args = ['-lgomp' ] link_args = ['-lgomp' ]
ext_mods = [Extension(name='GPy.kern.src.stationary_cython', try:
# So that we don't need numpy installed to determine it's a dependency.
import numpy as np
ext_mods = [Extension(name='GPy.kern.src.stationary_cython',
sources=['GPy/kern/src/stationary_cython.c', sources=['GPy/kern/src/stationary_cython.c',
'GPy/kern/src/stationary_utils.c'], 'GPy/kern/src/stationary_utils.c'],
include_dirs=[np.get_include(),'.'], include_dirs=[np.get_include(), '.'],
extra_compile_args=compile_flags, extra_compile_args=compile_flags,
extra_link_args = link_args), extra_link_args=link_args),
Extension(name='GPy.util.choleskies_cython', Extension(name='GPy.util.choleskies_cython',
sources=['GPy/util/choleskies_cython.c'], sources=['GPy/util/choleskies_cython.c'],
include_dirs=[np.get_include(),'.'], include_dirs=[np.get_include(), '.'],
extra_link_args = link_args, extra_link_args=link_args,
extra_compile_args=compile_flags), extra_compile_args=compile_flags),
Extension(name='GPy.util.linalg_cython', Extension(name='GPy.util.linalg_cython',
sources=['GPy/util/linalg_cython.c'], sources=['GPy/util/linalg_cython.c'],
include_dirs=[np.get_include(),'.'], include_dirs=[np.get_include(), '.'],
extra_compile_args=compile_flags, extra_compile_args=compile_flags,
extra_link_args = link_args), extra_link_args=link_args),
Extension(name='GPy.kern.src.coregionalize_cython', Extension(name='GPy.kern.src.coregionalize_cython',
sources=['GPy/kern/src/coregionalize_cython.c'], sources=['GPy/kern/src/coregionalize_cython.c'],
include_dirs=[np.get_include(),'.'], include_dirs=[np.get_include(), '.'],
extra_compile_args=compile_flags, extra_compile_args=compile_flags,
extra_link_args = link_args), extra_link_args=link_args),
Extension(name='GPy.models.state_space_cython', Extension(name='GPy.models.state_space_cython',
sources=['GPy/models/state_space_cython.c'], sources=['GPy/models/state_space_cython.c'],
include_dirs=[np.get_include(),'.'], include_dirs=[np.get_include(), '.'],
extra_compile_args=compile_flags, extra_compile_args=compile_flags,
extra_link_args = link_args)] extra_link_args=link_args)]
except ModuleNotFoundError:
ext_mods = []
setup(name = 'GPy', setup(name = 'GPy',
version = __version__, version = __version__,