diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2020-06-01 15:24:28 +0200 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2020-06-01 15:24:28 +0200 |
commit | 4aff8dc700a0790373d82ae24076359c09ee04c8 (patch) | |
tree | 7769285a8f5706e883cf0f56aadaa68e06b68f91 /src/python/setup.py.in | |
parent | c53567c85f936f78000471fcee6234e75f7742ca (diff) |
Interface for hera's bottleneck_distance
Diffstat (limited to 'src/python/setup.py.in')
-rw-r--r-- | src/python/setup.py.in | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/python/setup.py.in b/src/python/setup.py.in index b9f4e3f0..9b3c7521 100644 --- a/src/python/setup.py.in +++ b/src/python/setup.py.in @@ -48,10 +48,12 @@ ext_modules = cythonize(ext_modules) for module in pybind11_modules: my_include_dirs = include_dirs + [pybind11.get_include(False), pybind11.get_include(True)] - if module == 'hera': + if module == 'hera/wasserstein': my_include_dirs = ['@HERA_WASSERSTEIN_INCLUDE_DIR@'] + my_include_dirs + elif module == 'hera/bottleneck': + my_include_dirs = ['@HERA_BOTTLENECK_INCLUDE_DIR@'] + my_include_dirs ext_modules.append(Extension( - 'gudhi.' + module, + 'gudhi.' + module.replace('/', '.'), sources = [source_dir + module + '.cc'], language = 'c++', include_dirs = my_include_dirs, |