-
Notifications
You must be signed in to change notification settings - Fork 11
/
setup.py
81 lines (76 loc) · 2.33 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
"""kedro_airflow_k8s module."""
from setuptools import find_packages, setup
with open("README.md") as f:
README = f.read()
# Runtime Requirements.
INSTALL_REQUIRES = [
"kedro>=0.16, <0.18",
"click<8.0",
"semver~=2.10",
"python-slugify>=4.0.1",
"tabulate==0.8.9",
"pip-tools<6.11",
]
# Dev Requirements
EXTRA_REQUIRE = {
"tests": [
"pytest>=6.2.2, <7.0.0",
"pytest-cov>=2.8.0, <3.0.0",
"tox==3.21.1",
"pre-commit==2.9.3",
"apache-airflow[kubernetes]==2.1.0",
"apache-airflow-providers-cncf-kubernetes==1.1.0",
"apache-airflow-providers-apache-spark==2.0.1",
"mlflow-skinny==1.19.0",
"numpy<1.24", # mlflow 1.19 is incompatible with numpy >= 1.24
"sqlalchemy==1.3.23",
"responses>=0.13.0",
"kedro[spark.SparkDataSet]>=0.16, <0.18",
"WTForms<3.0.0",
"Markdown==3.3.4",
"protobuf==3.20.3",
],
"docs": [
"sphinx==3.4.2",
"recommonmark==0.8.2",
"sphinx_rtd_theme==0.5.2",
],
"gcp": [
"gcsfs<0.8.2",
],
"aws": ["s3fs>=0.6.1"],
"mlflow": ["kedro-mlflow==0.4.1"],
"spark": ["kedro[spark.SparkDataSet]>=0.16, <0.18"],
}
setup(
name="kedro-airflow-k8s",
version="0.8.2",
description="Kedro plugin with Airflow on Kubernetes support",
long_description=README,
long_description_content_type="text/markdown",
license="Apache Software License (Apache 2.0)",
python_requires=">=3",
classifiers=[
"Development Status :: 4 - Beta",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
],
keywords="kedro airflow kubernetes k8s ml mlops plugin",
author="Michal Zelechowski, Mariusz Strzelecki, Mateusz Pytel",
author_email="mateusz@getindata.com",
url="https://github.com/getindata/kedro-airflow-k8s/",
packages=find_packages(exclude=["ez_setup", "examples", "tests", "docs"]),
include_package_data=True,
zip_safe=False,
install_requires=INSTALL_REQUIRES,
extras_require=EXTRA_REQUIRE,
entry_points={
"kedro.project_commands": [
"airflow-k8s = kedro_airflow_k8s.cli:commands"
],
"kedro.hooks": [
"kubeflow_cfg_hook = "
"kedro_airflow_k8s.hooks:register_templated_config_loader",
],
},
)