diff options
| -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')])), |