forked from sintel-dev/Zephyr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
115 lines (100 loc) · 2.66 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
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""The setup script."""
from setuptools import setup, find_packages
try:
with open('README.md') as readme_file:
readme = readme_file.read()
except IOError:
readme = ''
try:
with open('HISTORY.md') as history_file:
history = history_file.read()
except IOError:
history = ''
install_requires = [
'numpy>=1.16.0,<1.23.0',
'pandas>=1,<2',
'composeml>=0.1.6,<0.10',
'featuretools>=1.0.0,<2.0.0',
'mlblocks>=0.5.0,<0.6',
'sigpro>=0.1.1.dev0',
'xgboost>=0.72.1,<1',
'jupyter==1.0.0',
]
setup_requires = [
'pytest-runner>=2.11.1',
]
tests_require = [
'pytest>=3.4.2',
'pytest-cov>=2.6.0',
'jupyter>=1.0.0,<2',
'rundoc>=0.4.3,<0.5',
]
development_requires = [
# general
'bumpversion>=0.5.3,<0.6',
'pip>=9.0.1',
'watchdog>=0.8.3,<0.11',
# docs
'm2r>=0.2.0,<0.3',
'nbsphinx>=0.5.0,<0.7',
'Sphinx>=1.7.1,<3',
'sphinx_rtd_theme>=0.2.4,<0.5',
'autodocsumm>=0.1.10',
'mistune>=0.7,<2',
'Jinja2<3.1',
# style check
'flake8>=3.7.7,<4',
'isort>=4.3.4,<5',
# fix style issues
'autoflake>=1.1,<2',
'autopep8>=1.4.3,<2',
# distribute on PyPI
'twine>=1.10.0,<4',
'wheel>=0.30.0',
# Advanced testing
'coverage>=4.5.1,<6',
'tox>=2.9.1,<4',
# Jupyter
'jupyter>=1.0.0',
]
setup(
author='MIT Data To AI Lab',
author_email='[email protected]',
classifiers=[
'Development Status :: 2 - Pre-Alpha',
'Intended Audience :: Developers',
'Natural Language :: English',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
],
description='Prediction engineering methods for Draco.',
entry_points={
'mlblocks': [
'primitives=zephyr_ml:MLBLOCKS_PRIMITIVES',
'pipelines=zephyr_ml:MLBLOCKS_PIPELINES'
],
},
extras_require={
'test': tests_require,
'dev': development_requires + tests_require,
},
install_package_data=True,
install_requires=install_requires,
long_description=readme + '\n\n' + history,
long_description_content_type='text/markdown',
include_package_data=True,
keywords='zephyr Draco Prediction Engineering',
name='zephyr-ml',
packages=find_packages(include=['zephyr_ml', 'zephyr_ml.*']),
python_requires='>=3.7,<3.9',
setup_requires=setup_requires,
test_suite='tests',
tests_require=tests_require,
url='https://github.com/sintel-dev/zephyr',
version='0.0.3.dev0',
zip_safe=False,
)