-
Notifications
You must be signed in to change notification settings - Fork 56
/
Copy pathsetup.py
48 lines (43 loc) · 1.4 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
from setuptools import setup
import sys
sys.path.append("./heat/core")
import version
print(version, dir(version))
with open("README.md", "r") as handle:
long_description = handle.read()
# with open('./heat/core/version.py') as handle:
# exec(handle.read())
# print(dir())
setup(
name="heat",
packages=[
"heat",
"heat.core",
"heat.core.cluster",
"heat.core.regression",
"heat.core.regression.lasso",
"heat.utils",
],
data_files=["README.md", "LICENSE"],
version=version.__version__,
description="A framework for high performance data analytics and machine learning.",
long_description=long_description,
long_description_content_type="text/markdown",
author="Helmholtz Association",
author_email="martin.siggel@dlr.de",
url="https://github.com/helmholtz-analytics/heat",
keywords=["data", "analytics", "tensors", "distributed", "gpu"],
classifiers=[
"Development Status :: 2 - Pre-Alpha",
"Programming Language :: Python :: 3.5",
"License :: OSI Approved :: MIT License",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering",
],
install_requires=["mpi4py>=3.0.0", "numpy>=1.13.0", "torch>=1.3.0"],
extras_require={
"hdf5": ["h5py>=2.8.0"],
"netcdf": ["netCDF4>=1.4.0,<=1.5.2"],
"dev": ["pre-commit>=1.18.3"],
},
)