ml-pipelines-sdk==1.21.0
absl-py<2.0.0,>=0.9
ml-metadata<1.22.0,>=1.21.0
packaging>=22
portpicker<2,>=1.3.1
protobuf<7.0.0,>=6.0.0
docker<8,>=7
google-apitools<1,>=0.5
google-api-python-client<2,>=1.8
jinja2<4,>=2.7.3
typing-extensions<5
apache-beam[gcp]<3,>=2.47
attrs<26,>=19.3.0
click<9,>=7
google-api-core<3
google-cloud-aiplatform<2,>=1.6.2
google-cloud-bigquery<4,>=3
grpcio<2,>=1.28.1
keras-tuner!=1.4.0,!=1.4.1,<2,>=1.0.4
kubernetes<27,>=10.0.1
numpy<3,>=1.16
orjson!=3.10.7
scipy<2
scikit-learn==1.5.2
pyyaml<7,>=6
tensorflow<2.22,>=2.21.0
tensorflow-hub<0.16,>=0.15.0
tensorflow-data-validation<1.22.0,>=1.21.0
tensorflow-model-analysis<0.53.0,>=0.52.0
tensorflow-serving-api<2.22,>=2.19.1
tensorflow-transform<1.22.0,>=1.21.0
tfx-bsl<1.22.0,>=1.21.0

[:python_version < "3.13"]
pyarrow<19,>=10

[:python_version >= "3.13"]
pyarrow<19,>=18

[airflow]

[airflow:python_version < "3.13"]
apache-airflow[mysql]<3,>=1.10.14

[all]
pytest<=8,>=5
pytest-subtests==0.13.1
flatbuffers>=1.12
struct2tensor<0.53.0,>=0.52.0
presto-python-client<0.8,>=0.7
slackclient<3,>=2.8.2
websocket-client<1,>=0.57
scikit-learn<2,>=1.0
tensorflow-cloud<0.2,>=0.1

[all:python_version < "3.12"]
kfp<2.17.0,>=2.6.0
kfp-pipeline-spec<2.17.0,>=0.3.0

[all:python_version < "3.13"]
apache-airflow[mysql]<3,>=1.10.14
jax<0.4.24
jaxlib<0.4.24
flax<1
optax<1

[docker-image]
mmh<3,>=2.2
python-snappy>=0.7
tensorflow-cloud<0.2,>=0.1
tensorflow-io<=0.24.0,>=0.9.0

[docker-image:python_version < "3.12"]
kfp<2.17.0,>=2.6.0
kfp-pipeline-spec<2.17.0,>=0.3.0

[docs]
mkdocs
mkdocstrings[python]
mkdocs-material
griffe-inherited-docstrings
mkdocs-autorefs
mkdocs-jupyter
mkdocs-caption
markdown-grid-tables

[examples]
presto-python-client<0.8,>=0.7
slackclient<3,>=2.8.2
websocket-client<1,>=0.57
scikit-learn<2,>=1.0
tensorflow-cloud<0.2,>=0.1

[flax]

[flax:python_version < "3.13"]
jax<0.4.24
jaxlib<0.4.24
flax<1
optax<1

[kfp]

[kfp:python_version < "3.12"]
kfp<2.17.0,>=2.6.0
kfp-pipeline-spec<2.17.0,>=0.3.0

[test]
pytest<=8,>=5
pytest-subtests==0.13.1

[test:python_version < "3.12"]
kfp<2.17.0,>=2.6.0
kfp-pipeline-spec<2.17.0,>=0.3.0

[test:python_version < "3.13"]
apache-airflow[mysql]<3,>=1.10.14

[tf-ranking]
struct2tensor<0.53.0,>=0.52.0

[tfdf]

[tfjs]

[tflite-support]
flatbuffers>=1.12
