absl-py<2.0.0,>=0.9
apache-beam[gcp]<3,>=2.53
numpy>=1.22.0
protobuf<7.0.0,>=6.0.0
pyarrow>14
pydot<2,>=1.2
tensorflow<2.22,>=2.21
tensorflow-metadata<1.22.0,>=1.21.0
tf_keras>=2
tfx-bsl<1.22.0,>=1.21.0

[dev]
pre-commit

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

[test]
pytest>=8.0
