pyro-risks/setup.py at master · Rim-chan/pyro-risks · GitHub

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
# Copyright (C) 2021, Pyronear contributors.

# This program is licensed under the GNU Affero General Public License version 3.
# See LICENSE or go to <https://www.gnu.org/licenses/agpl-3.0.txt> for full license details.

#!usr/bin/python

"""
Package installation setup
"""

import os
import subprocess
from setuptools import setup, find_packages

package_name = "pyro_risks"
with open(os.path.join(package_name, "version.py")) as version_file:
    version = version_file.read().strip()
sha = "Unknown"

cwd = os.path.dirname(os.path.abspath(__file__))

try:
    sha = (
        subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=cwd)
        .decode("ascii")
        .strip()
    )
except Exception:
    pass

if os.getenv("BUILD_VERSION"):
    version = os.getenv("BUILD_VERSION")
elif sha != "Unknown":
    version += "+" + sha[:7]
print("Building wheel {}-{}".format(package_name, version))


with open("README.md") as f:
    readme = f.read()

requirements = [
    "pandas>=1.1.4",
    "geopandas>=0.8.1",
    "Rtree>=0.9.4",
    "Shapely>=1.7.1",
    "netCDF4>=1.5.4",
    "requests>=2.24.0",
    "xarray>=0.16.1",
    "scipy>=1.5.4",
    "scikit-learn>=0.23.2",
    "imbalanced-learn>=0.7.0",
    "xgboost==1.2.1",
    "xlrd==1.2.0",
    "numpy>=1.18.5",
    "cdsapi==0.4.0",
    "python-dotenv>=0.15.0",
    "plot-metric==0.0.6",
    "dvc>=2.0.5",
    "dvc[gdrive]>=2.0.5",
]

setup(
    name=package_name,
    version=version,
    author="Pyronear Contributors",
    description="Pre-processing pipelines and models for wildfire forecasting and monitoring",
    long_description=readme,
    long_description_content_type="text/markdown",
    url="https://github.com/pyronear/pyro-risks",
    download_url="https://github.com/pyronear/pyro-risks/tags",
    license="GPLv3",
    entry_points={"console_scripts": ["pyrorisks = pyro_risks.main:main"]},
    classifiers=[
        "Development Status :: 2 - Pre-Alpha",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
        "Natural Language :: English",
        "Operating System :: OS Independent",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Topic :: Scientific/Engineering",
        "Topic :: Scientific/Engineering :: Mathematics",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Topic :: Software Development",
        "Topic :: Software Development :: Libraries",
        "Topic :: Software Development :: Libraries :: Python Modules",
    ],
    keywords=["data science", "time series", "machine learning"],
    packages=find_packages(exclude=("test",)),
    zip_safe=True,
    python_requires=">=3.6.0",
    include_package_data=True,
    install_requires=requirements,
    package_data={"": ["LICENSE"]},
)