-
Notifications
You must be signed in to change notification settings - Fork 85
/
setup.py
84 lines (80 loc) · 2.18 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from setuptools import setup
extras_require = {
'shellcomplete': ['click_completion'],
'tensorflow': [
'tensorflow~=2.2,>=2.2.1,!=2.3.0', # c.f. https://github.com/tensorflow/tensorflow/pull/40789
'tensorflow-probability~=0.10,>=0.10.1',
],
'torch': ['torch~=1.8'],
'jax': ['jax~=0.2.8', 'jaxlib~=0.1.58,!=0.1.68'], # c.f. Issue 1501
'xmlio': ['uproot>=4.1.1'],
'minuit': ['iminuit>=2.4'],
}
extras_require['backends'] = sorted(
set(
extras_require['tensorflow']
+ extras_require['torch']
+ extras_require['jax']
+ extras_require['minuit']
)
)
extras_require['contrib'] = sorted({'matplotlib', 'requests'})
extras_require['lint'] = sorted({'flake8', 'black'})
extras_require['test'] = sorted(
set(
extras_require['backends']
+ extras_require['xmlio']
+ extras_require['contrib']
+ extras_require['shellcomplete']
+ [
'pytest~=6.0',
'pytest-cov>=2.5.1',
'pytest-mock',
'pytest-benchmark[histogram]',
'pytest-console-scripts',
'pytest-mpl',
'pydocstyle',
'papermill~=2.0',
'nteract-scrapbook~=0.2',
'jupyter',
'graphviz',
]
)
)
extras_require['docs'] = sorted(
set(
extras_require['xmlio']
+ extras_require['contrib']
+ [
'sphinx>=4.0.0',
'sphinxcontrib-bibtex~=2.1',
'sphinx-click',
'sphinx_rtd_theme',
'nbsphinx',
'ipywidgets',
'sphinx-issues',
'sphinx-copybutton>=0.3.2',
]
)
)
extras_require['develop'] = sorted(
set(
extras_require['docs']
+ extras_require['lint']
+ extras_require['test']
+ [
'nbdime',
'bump2version',
'ipython',
'pre-commit',
'check-manifest',
'codemetapy>=0.3.4',
'twine',
]
)
)
extras_require['complete'] = sorted(set(sum(extras_require.values(), [])))
setup(
extras_require=extras_require,
use_scm_version=lambda: {'local_scheme': lambda version: ''},
)