-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
63 lines (56 loc) · 1.59 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
import logging
from setuptools import setup, find_packages
test_deps = [
'pytest>=4',
'pytest-cov>=2.6.0',
'pytest-flake8'
]
try:
import mxnet
mxnet_requires = []
except ModuleNotFoundError:
mxnet_requires = ["mxnet"]
except Exception as e:
mxnet_requires = []
logging.error(e)
try:
import torch
ml_pytorch_deps = []
except ModuleNotFoundError:
import sys
if 5 <= sys.version_info[1]:
ml_pytorch_deps = ["torch"]
else:
ml_pytorch_deps = []
logging.warning("Current python version %s is not supported by pytorch", str(sys.version_info[:2]))
except Exception as e:
ml_pytorch_deps = []
logging.error(e)
setup(
name='PyBaize',
version='0.0.7',
extras_require={
'test': test_deps + mxnet_requires + ml_pytorch_deps,
'mxnet': mxnet_requires,
'torch': ml_pytorch_deps,
'full': mxnet_requires + ml_pytorch_deps
},
packages=find_packages(),
install_requires=[
'longling[ml]>=1.3.35',
'nni>=1.8'
], # And any other dependencies foo needs
entry_points={
},
classifiers=[
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
"Environment :: Other Environment",
"Intended Audience :: Developers",
"License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
"Operating System :: OS Independent",
"Topic :: Software Development :: Libraries :: Python Modules",
],
)