|
62 | 62 | url="https://github.com/PythonOT/POT", |
63 | 63 | packages=find_packages(exclude=["benchmarks"]), |
64 | 64 | 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 | + ] |
83 | 85 | ), |
84 | 86 | platforms=["linux", "macosx", "windows"], |
85 | 87 | download_url="https://github.com/PythonOT/POT/archive/{}.tar.gz".format( |
|
0 commit comments