-
Notifications
You must be signed in to change notification settings - Fork 10
/
setup_docker.py
124 lines (121 loc) · 3.43 KB
/
setup_docker.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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# License: Apache-2.0
import setuptools
import numpy
from Cython.Build import cythonize
from gators import __version__
import pyximport
pyximport.install()
extensions = [
setuptools.Extension(
'imputer',
['/gators/gators/imputers/imputer.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'encoder',
['/gators/gators/encoders/encoder.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'data_cleaning',
['/gators/gators/data_cleaning/data_cleaning.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'binning',
['/gators/gators/binning/binning.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'clipping',
['/gators/gators/clipping/clipping.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'scaler',
['/gators/gators/scalers/scaler.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'feature_gen',
['/gators/gators/feature_generation/feature_gen.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'feature_gen_dt',
['/gators/gators/feature_generation_dt/feature_gen_dt.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
),
setuptools.Extension(
'feature_gen_str',
['/gators/gators/feature_generation_str/feature_gen_str.pyx'],
include_dirs=[numpy.get_include()],
extra_compile_args=['-O3'],
)
]
setuptools.setup(
name='gators',
version=__version__,
author='Simility Data Team',
author_email='cpoli@paypal.com',
options={'bdist_wheel': {'universal': True}},
description='Model building and Model deployment library',
packages=setuptools.find_packages(exclude=['examples', 'doc']),
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: Apache-2 License',
'Operating System :: OS Independent',
],
python_requires='>=3.6',
license='Apache-2.0 Software License',
setup_requires=['numpy', 'Cython'],
install_requires=[
'setuptools>=41.0.0',
'numpy==1.19.5',
'requests>=2.23.0',
'tqdm>=4.43.0',
'scipy>=1.5.2',
'Cython>=0.29.21',
'dill>=0.3.1.1',
'scikit-learn==0.23.1',
'seaborn>=0.11.0',
'pandas>=0.25.3<1.2',
'treelite>=0.93',
'treelite-runtime>=0.93',
'xgboost>=0.90',
'pyDOE>=0.3.8',
'scikit-optimize>=0.8.1',
'emcee>=3.0.2',
'pyspark>=2.4.3',
'koalas',
'hyperopt>=0.2.5',
'Lightgbm',
'pytest>=5.0.0',
'pytest-cov>=2.6.0',
'pylama>=7.6.5',
'sphinx>=3.3.0',
'nbsphinx>=0.8.0',
'pydata_sphinx_theme',
'ipykernel',
'jupyter',
'numpydoc',
'tox',
'tox-wheel',
],
package_data={'gators': [
'*.c',
]},
include_package_data=True,
ext_modules=cythonize(
extensions,
compiler_directives={'language_level': "3"}),
zip_safe=False,
)