diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-03-21 09:32:06 +0100 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-03-21 09:32:06 +0100 |
commit | 0bbd36c0b569e58c3d47177e816b3541fac4b916 (patch) | |
tree | 2cba51d5447afa3356470371adb216b587832678 | |
parent | a84f2c3e23edd1fa89975bd77b08672f518d5ca4 (diff) |
cleanupt cpp wrapper name
-rw-r--r-- | ot/lp/__init__.py | 2 | ||||
-rw-r--r-- | ot/lp/emd_wrap.pyx (renamed from ot/lp/emd.pyx) | 0 | ||||
-rwxr-xr-x | setup.py | 4 |
3 files changed, 3 insertions, 3 deletions
diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py index 5674cf6..2e7c52e 100644 --- a/ot/lp/__init__.py +++ b/ot/lp/__init__.py @@ -5,7 +5,7 @@ Solvers for the original linear program OT problem import numpy as np # import compiled emd -from .emd import emd_c, emd2_c +from .emd_wrap import emd_c, emd2_c from ..utils import parmap import multiprocessing diff --git a/ot/lp/emd.pyx b/ot/lp/emd_wrap.pyx index 46794ab..46794ab 100644 --- a/ot/lp/emd.pyx +++ b/ot/lp/emd_wrap.pyx @@ -38,8 +38,8 @@ setup(name='POT', url='https://github.com/rflamary/POT', packages=find_packages(), ext_modules = cythonize(Extension( - "ot.lp.emd", # the extension name - sources=["ot/lp/emd.pyx", "ot/lp/EMD_wrap.cpp"], # the Cython source and + "ot.lp.emd_wrap", # the extension name + sources=["ot/lp/emd_wrap.pyx", "ot/lp/EMD_wrap.cpp"], # the Cython source and # additional C++ source files language="c++", # generate and compile C++ code, include_dirs=[numpy.get_include(),os.path.join(ROOT,'ot/lp')])), |