summaryrefslogtreecommitdiff
path: root/src/python/setup.py.in
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-06-09 19:45:31 +0200
committerGitHub <noreply@github.com>2020-06-09 19:45:31 +0200
commit58f7cd2d1fec9b12665ad10c16b9812eba303570 (patch)
treed68fd29693505e78811db0ff1b9fe052fc11637b /src/python/setup.py.in
parent434cc66fbf832ca492e5f0710c8afc67d681637a (diff)
parentdacc3e363ded8b68bb4b71c1602e2c52b10b36e5 (diff)
Merge pull request #337 from mglisse/herab
Interface for hera's bottleneck_distance
Diffstat (limited to 'src/python/setup.py.in')
-rw-r--r--src/python/setup.py.in6
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,