Skip to content

Commit c36a45d

Browse files
committed
bugfix in cythonize
1 parent aef4b55 commit c36a45d

File tree

1 file changed

+20
-18
lines changed

1 file changed

+20
-18
lines changed

setup.py

Lines changed: 20 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -62,24 +62,26 @@
6262
url="https://github.com/PythonOT/POT",
6363
packages=find_packages(exclude=["benchmarks"]),
6464
ext_modules=cythonize(
65-
Extension(
66-
name="ot.lp.emd_wrap",
67-
sources=[
68-
"ot/lp/emd_wrap.pyx",
69-
"ot/lp/EMD_wrapper.cpp",
70-
], # cython/c++ src files
71-
language="c++",
72-
include_dirs=[numpy.get_include(), os.path.join(ROOT, "ot/lp")],
73-
extra_compile_args=compile_args,
74-
extra_link_args=link_args,
75-
),
76-
Extension(
77-
name="ot.partial.partial_wrap",
78-
sources=["partial_wrap.pyx"],
79-
include_dirs=[numpy.get_include(), os.path.join(ROOT, "ot/partial")],
80-
extra_compile_args=["-O3"],
81-
language="c++"
82-
)
65+
[
66+
Extension(
67+
name="ot.lp.emd_wrap",
68+
sources=[
69+
"ot/lp/emd_wrap.pyx",
70+
"ot/lp/EMD_wrapper.cpp",
71+
], # cython/c++ src files
72+
language="c++",
73+
include_dirs=[numpy.get_include(), os.path.join(ROOT, "ot/lp")],
74+
extra_compile_args=compile_args,
75+
extra_link_args=link_args,
76+
),
77+
Extension(
78+
name="ot.partial.partial_wrap",
79+
sources=["partial_wrap.pyx"],
80+
include_dirs=[numpy.get_include(), os.path.join(ROOT, "ot/partial")],
81+
extra_compile_args=["-O3"],
82+
language="c++"
83+
)
84+
]
8385
),
8486
platforms=["linux", "macosx", "windows"],
8587
download_url="https://github.com/PythonOT/POT/archive/{}.tar.gz".format(

0 commit comments

Comments
 (0)