forked from ploomber/sklearn-evaluation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
84 lines (76 loc) · 1.98 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
import re
import ast
from setuptools import setup, find_packages
from glob import glob
from os.path import splitext
from os.path import basename
_version_re = re.compile(r'__version__\s+=\s+(.*)')
with open('src/sklearn_evaluation/__init__.py', 'rb') as f:
VERSION = str(
ast.literal_eval(
_version_re.search(f.read().decode('utf-8')).group(1)))
DOWNLOAD_URL = ('https://github.com/edublancas/sklearn-evaluation/tarball/{}'.
format(VERSION))
DOCS = [
'sphinx',
'sphinx-rtd-theme',
'ploomber',
'nbsphinx',
'seaborn',
# to display progress bar when executing notebooks using papermill
# in NotebookCollection.py example
'ipywidgets'
]
TEST = [
'jupytext',
'papermill',
'ipykernel',
'pytest',
# need to pin this version because pytest 4 breaks matplotlib image
# comparison tests
'pytest-cov',
# TODO: update config so coveralls 3 works
'coveralls<3',
]
DEV = [
'flake8',
'yapf',
]
ALL = DOCS + TEST + DEV
setup(
name='sklearn-evaluation',
packages=find_packages('src'),
package_dir={'': 'src'},
py_modules=[splitext(basename(path))[0] for path in glob('src/*.py')],
version=VERSION,
description=('scikit-learn model evaluation made easy: plots, tables and'
'markdown reports.'),
url='http://github.com/edublancas/sklearn-evaluation',
download_url=DOWNLOAD_URL,
author='Eduardo Blancas Reyes',
author_email='[email protected]',
license='MIT',
keywords=['datascience', 'machinelearning'],
classifiers=[],
include_package_data=True,
install_requires=[
# compute metrics
'scikit-learn',
# plotting
'matplotlib',
# misc
'decorator',
# metric tables
'tabulate',
'jinja2',
# reports
'mistune',
'pandas',
'nbformat',
# notebook compare
'ipython',
'black',
],
extras_require={
'all': ALL,
})