-
Notifications
You must be signed in to change notification settings - Fork 11
/
pyproject.toml
80 lines (70 loc) · 2.72 KB
/
pyproject.toml
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
[tool.poetry]
name = "kedro-vertexai"
version = "0.11.1"
description = "Kedro plugin with GCP Vertex AI support"
readme = "README.md"
authors = ['Marcin Zabłocki <[email protected]>', 'Mateusz Pytel <[email protected]>', 'Mariusz Strzelecki <[email protected]>', 'Artur Dobrogowski <[email protected]>']
maintainers = ['GetInData MLOPS <[email protected]>']
homepage = "https://github.com/getindata/kedro-vertexai"
repository = "https://github.com/getindata/kedro-vertexai"
documentation = "https://kedro-vertexai.readthedocs.io/"
keywords = ['kedro-plugin', 'kedro', 'mlops', 'vertexai', 'googlecloudplatform', 'machinelearning']
license = "Apache-2.0"
classifiers = [
"Development Status :: 4 - Beta", # license and python versions added automatically
]
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.coverage.run]
branch = true
relative_files = true
[tool.coverage.report]
exclude_lines = [
"pragma: no cover",
"raise NotImplementedError"
]
[tool.isort]
known_third_party = ["click","google","kedro","kfp","kubernetes","tabulate", "pydantic","semver","setuptools"]
[tool.poetry.dependencies]
python = ">=3.8,<3.11" # weird error OverrideNeeded when bumping up. Could be related to old versions with kfp
kedro = ">=0.19.0,<0.20"
click = ">=8.0.4"
kfp = ">=2.0.0,<3.0"
tabulate = ">=0.8.7"
semver = ">=2.10,<4.0.0"
toposort = ">1.0,<2.0"
pyarrow = ">=14.0.1" # Stating explicitly for sub-dependency due to critical vulnerability
pydantic = ">=2,<3"
google-auth = "<3"
google-cloud-scheduler = ">=2.3.2"
google-cloud-iam = "<3"
gcsfs = ">=2022.1"
fsspec = ">=2022.1"
google-cloud-storage = "<3.0.0"
grpcio = ">=1.4.0,<2.0.0"
grpcio-status = ">=1.4.0,<2.0.0"
protobuf = ">=3.18.0,<21.0"
kedro-mlflow = {version = ">=0.12.1,<0.13", optional = true}
cachetools = ">=3.0,<6.0"
# pyaml in version 5 does problems with installing binaries/wheel in cicd env with python 3.10. The following fixes that:
# pyyaml = ">=6.0,<7"
google-cloud-aiplatform = {extras = ["metadata"], version = "^1.59.0"}
cloudpickle = "^3.0.0"
mlflow = "^2.14.3"
[tool.poetry.extras]
mlflow = ["kedro-mlflow"]
[tool.poetry.dev-dependencies]
pytest = "<=8.0.2"
pytest-cov = ">=2.8.0, <4.0.0"
tox = ">=3.25.1,<4.0.0"
pre-commit = "2.20.0"
pytest-subtests = ">=0.8.0, <1.0.0"
responses = ">=0.13.4"
[tool.poetry.plugins] # Optional super table
[tool.poetry.plugins."kedro.project_commands"]
"vertexai" = "kedro_vertexai.cli:commands"
[tool.poetry.plugins."kedro.hooks"]
"vertexai_mlflow_tags_hook" = "kedro_vertexai.hooks:mlflow_tags_hook"
[tool.poetry.plugins."mlflow.request_header_provider"]
"unused" = "kedro_vertexai.auth.mlflow_request_header_provider:DynamicMLFlowRequestHeaderProvider"