forked from sql-machine-learning/elasticdl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
36 lines (33 loc) · 1.06 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
from setuptools import find_packages, setup
with open("elasticdl/requirements.txt") as f:
required_deps = f.read().splitlines()
extras = {}
with open("elasticdl/requirements-dev.txt") as f:
extras["develop"] = f.read().splitlines()
setup(
name="elasticdl",
version="develop",
description="A Kubernetes-native Deep Learning Framework",
long_description="ElasticDL is a Kubernetes-native deep learning framework"
" built on top of TensorFlow 2.0 that supports"
" fault-tolerance and elastic scheduling.",
long_description_content_type="text/markdown",
author="Ant Financial",
url="https://elasticdl.org",
install_requires=required_deps,
extras_require=extras,
python_requires=">=3.5",
packages=find_packages(exclude=["*test*"]),
package_data={
"": [
"proto/*.proto",
"docker/*",
"Makefile",
"requirements.txt",
"pkg/kernel/capi/*",
]
},
entry_points={
"console_scripts": ["elasticdl=elasticdl.python.elasticdl.client:main"]
},
)