forked from SMTorg/smt
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
47 lines (43 loc) · 1.34 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
from numpy.distutils.core import setup, Extension
import os
from subprocess import call
if os.name == 'nt':
# If OS is Windows, don't compile Fortran code
ext = []
else:
# If OS is OS X or Linux, assume Fortran compilers are available
ext = [
Extension(name='smt.IDWlib', extra_compile_args=['-fbounds-check'],
sources=['src_f/idw.f95']),
Extension(name='smt.RBFlib', extra_compile_args=['-fbounds-check'],
sources=['src_f/rbf.f95']),
Extension(name='smt.RMTSlib', extra_compile_args=['-fbounds-check'],
sources=['src_f/rmts.f95', 'src_f/utils.f95']),
Extension(name='smt.RMTBlib', extra_compile_args=['-fbounds-check'],
sources=['src_f/rmtb.f95', 'src_f/utils.f95']),
]
setup(name='smt',
version='0.1',
description='The Surrogate Model Toolbox (SMT)',
author='Mohamed Amine Bouhlel',
author_email='mbouhlel@umich.edu',
license='BSD-3',
packages=[
'smt',
'smt/problems',
'smt/sampling',
'smt/utils',
],
install_requires=[
'scikit-learn',
'pyDOE',
'pyamg',
],
zip_safe=False,
ext_modules=ext,
)
for lib_name in ['IDWlib', 'RBFlib', 'RMTSlib', 'RMTBlib']:
try:
call(['mv', 'smt/%s.cpython-36m-darwin.so' % lib_name, 'smt/%s.so' % lib_name])
except:
pass