diff options
| author | Marc Glisse <marc.glisse@inria.fr> | 2020-06-09 19:45:31 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-06-09 19:45:31 +0200 |
| commit | 58f7cd2d1fec9b12665ad10c16b9812eba303570 (patch) | |
| tree | d68fd29693505e78811db0ff1b9fe052fc11637b /src/python/setup.py.in | |
| parent | 434cc66fbf832ca492e5f0710c8afc67d681637a (diff) | |
| parent | dacc3e363ded8b68bb4b71c1602e2c52b10b36e5 (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.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, |