click>=7.0
tensorflow-datasets<v4.9.0,>=1.3.0
typeguard>=3.0
protobuf<3.21

[tensorflow]
tensorflow>=2.8.4

[tensorflow_gpu]
tensorflow-gpu>=2.8.4

[test]
pytype==2024.10.11
pytest==9.0.3
pytest-cov==7.1.0
ruff==0.15.13
