forked from Trusted-AI/AIX360
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
53 lines (50 loc) · 1.53 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
import setuptools
version = '0.2.1'
with open("aix360/version.py", 'w') as f:
f.write('# generated by setup.py\nversion = "{}"\n'.format(version))
setuptools.setup(
name='aix360',
version=version,
description='IBM AI Explainability 360',
authos='aix360 developers',
url='https://github.com/IBM/AIX360',
author_email='aix360@us.ibm.com',
packages=setuptools.find_packages(),
license='Apache License 2.0',
long_description=open('README.md', 'r', encoding='utf-8').read(),
long_description_content_type='text/markdown',
install_requires=[
'joblib>=0.11',
'scikit-learn>=0.21.2',
'torch',
'torchvision',
'cvxpy>==1.1',
'Image',
'tensorflow==1.14',
'h5py<3.0.0',
'keras==2.3.1',
'matplotlib',
'numpy',
'pandas',
'scipy>=0.17',
'xport',
'scikit-image',
'requests',
'xgboost==1.1.0',
'bleach>=2.1.0',
'docutils>=0.13.1',
'Pygments',
'osqp',
'lime==0.1.1.37',
'shap==0.34.0',
'nyoka==5.2.0',
'xmltodict==0.12.0',
'numba',
'tqdm',
'kaggle',
'otoc @ git+https://github.com/IBM/otoc@main#egg=otoc'
],
package_data={'aix360': ['data/*', 'data/*/*', 'data/*/*/*', 'models/*', 'models/*/*', 'models/*/*/*']},
include_package_data=True,
zip_safe=False
)