-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
64 lines (60 loc) · 3 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
from setuptools import setup, Extension, find_packages
from Cython.Build import cythonize
from pathlib import Path
import scipy
import numpy
import pybind11
# import Cython
# print(Cython.__version__)
numpy_inc = numpy.get_include()
pybind11_inc = pybind11.get_include()
# cython扩展模块
cython_extensions = [
Extension("UQPyL.surrogates.mars.core._types", ["UQPyL/surrogates/mars/core/_types.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._util", ["UQPyL/surrogates/mars/core/_util.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._forward", ["UQPyL/surrogates/mars/core/_forward.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._record", ["UQPyL/surrogates/mars/core/_record.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._basis", ["UQPyL/surrogates/mars/core/_basis.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._pruning", ["UQPyL/surrogates/mars/core/_pruning.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._qr", ["UQPyL/surrogates/mars/core/_qr.pyx"], include_dirs=[numpy_inc]),
Extension("UQPyL.surrogates.mars.core._knot_search", ["UQPyL/surrogates/mars/core/_knot_search.pyx"], include_dirs=[numpy_inc]),
Extension('UQPyL.surrogates.regression.lasso', ["UQPyL/surrogates/regression/lasso/lasso_fast.pyx"], include_dirs=[numpy_inc])
]
#pybind11扩展模块
pybind11_extensions = [
Extension("UQPyL.surrogates.svr.core", [str(Path("UQPyL/surrogates/svr/core/libsvm_interface.cpp")), str(Path("UQPyL/surrogates/svr_/svm.cpp"))], include_dirs=[numpy_inc, pybind11_inc]),
]
extensions=cythonize(cython_extensions, compiler_directives={'cdivision': True, 'boundscheck': False})+pybind11_extensions
# 使用cythonize编译扩展模块
# setup(
# name="UQPyL",
# description='A package for uncertainty quantification and parameter optimization with surrogate models.',
# version="2.0.0",
# author="wmtSky",
# url='https://https://github.com/smasky/UQPyL',
# ext_modules=extensions,
# )
setup(
name="UQPyL",
author="wmtSky",
version="2.0.5",
author_email="wmtsky@hhu.edu.cn",
# ... 其他常规setup参数 ...
ext_modules=extensions, # 如果有自定义的编译行为
packages=find_packages(),
description="A python package for parameter uncertainty quantification and optimization",
long_description=open("README.md").read(),
long_description_content_type="text/markdown", # 如果是Markdown格式
classifiers=[
# 添加适合的类目,例如
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent',
],
)