diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2020-02-12 12:44:51 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-02-12 12:44:51 +0100 |
commit | bed30b19e57669c0b8ad385f1124586ed3499a2d (patch) | |
tree | 6ddbe2f3015899159818bd6e3003d78dd920d707 /src/python/setup.py.in | |
parent | ee0f12f1df406c81c6ad860c494eed908021fad9 (diff) | |
parent | d6f3165831d20bf3a91f1ff7e9734a574eaa567a (diff) |
Merge pull request #182 from mglisse/ext
Interface to hera's Wasserstein distance
Diffstat (limited to 'src/python/setup.py.in')
-rw-r--r-- | src/python/setup.py.in | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/src/python/setup.py.in b/src/python/setup.py.in index bd7fb180..f968bd59 100644 --- a/src/python/setup.py.in +++ b/src/python/setup.py.in @@ -12,6 +12,7 @@ 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,6 +43,18 @@ 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=find_packages(), # find_namespace_packages(include=["gudhi*"]) @@ -49,7 +62,7 @@ setup( 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',], ) |