diff options
| author | mathieu <mathieu.carriere3@gmail.com> | 2020-02-13 15:33:22 -0500 |
|---|---|---|
| committer | mathieu <mathieu.carriere3@gmail.com> | 2020-02-13 15:33:22 -0500 |
| commit | 2f2db197a38e45ac4fe01dec0c029171c251029b (patch) | |
| tree | 5c6522af65647afeb2f7e5e4b0c2858ee2352679 /src/python/setup.py.in | |
| parent | d21640a16113a3c56389efcb060b3430af9f256d (diff) | |
| parent | bed30b19e57669c0b8ad385f1124586ed3499a2d (diff) | |
Merge branch 'master' of https://github.com/GUDHI/gudhi-devel into wasserstein_representations
Diffstat (limited to 'src/python/setup.py.in')
| -rw-r--r-- | src/python/setup.py.in | 21 |
1 files changed, 17 insertions, 4 deletions
diff --git a/src/python/setup.py.in b/src/python/setup.py.in index f993165c..f968bd59 100644 --- a/src/python/setup.py.in +++ b/src/python/setup.py.in @@ -8,10 +8,11 @@ - YYYY/MM Author: Description of the modification """ -from setuptools import setup, Extension +from setuptools import setup, Extension, find_packages from Cython.Build import cythonize from numpy import get_include as numpy_get_include import sys +import pybind11 __author__ = "Vincent Rouvreau" __copyright__ = "Copyright (C) 2016 Inria" @@ -42,14 +43,26 @@ for module in modules: runtime_library_dirs=runtime_library_dirs, cython_directives = {'language_level': str(sys.version_info[0])},)) +ext_modules = cythonize(ext_modules) + +ext_modules.append(Extension( + 'gudhi.hera', + sources = [source_dir + 'hera.cc'], + language = 'c++', + include_dirs = include_dirs + + ['@HERA_WASSERSTEIN_INCLUDE_DIR@', + pybind11.get_include(False), pybind11.get_include(True)], + extra_compile_args=extra_compile_args + [@GUDHI_PYBIND11_EXTRA_COMPILE_ARGS@], + )) + setup( name = 'gudhi', - packages=["gudhi","gudhi.representations"], + packages=find_packages(), # find_namespace_packages(include=["gudhi*"]) author='GUDHI Editorial Board', author_email='gudhi-contact@lists.gforge.inria.fr', version='@GUDHI_VERSION@', url='http://gudhi.gforge.inria.fr/', - ext_modules = cythonize(ext_modules), + ext_modules = ext_modules, install_requires = ['cython','numpy >= 1.9',], - setup_requires = ['numpy >= 1.9',], + setup_requires = ['numpy >= 1.9','pybind11',], ) |