click>=7.0
tensorflow-datasets<v4.9.0,>=1.3.0
typeguard<3.0.0,>=2.5.1
protobuf<3.20

[:python_version < "3.8"]
importlib-metadata~=2.0

[tensorflow]
tensorflow>=1.14.0

[tensorflow_gpu]
tensorflow-gpu>=1.14.0

[test]
black==23.7.0
docformatter==1.7.4
flake8==6.0.0
isort==5.12.0
pytest==7.4.0
pytest-cov==4.1.0
pytype==2023.6.16
