-
Notifications
You must be signed in to change notification settings - Fork 5
/
setup.py
executable file
·49 lines (41 loc) · 1.04 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
from setuptools import setup, find_packages
with open("README.md", "r") as fh:
long_description = fh.read()
install_requires = [
"tensorboard",
"numpy",
"matplotlib",
"pillow",
"xarray",
"zarr",
"netcdf4",
"pytorch-lightning",
"earthnet",
"segmentation-models-pytorch"
]
scripts = [
"scripts/debug.py",
"scripts/train.py",
"scripts/test.py",
"scripts/tune.py",
"demo/app.py",
"scripts/slurmrun.sh",
]
setup(
name="earthnet-models-pytorch",
version="0.1.0",
description="EarthNet Models PyTorch",
author="Vitus Benson, Claire Robin",
author_email="{vbenson, crobin}@bgc-jena.mpg.de",
url="https://earthnet.tech",
long_description=long_description,
long_description_content_type="text/markdown",
classifiers=[
"Intended Audience :: Science/Research",
"License :: Other/Proprietary License",
"Programming Language :: Python :: 3",
],
packages=find_packages(),
scripts=scripts,
install_requires=install_requires,
)