requirements-test.txt

# Tools for static checking.
black == 19.10b0
flake8 == 3.7.9
isort == 5.6.4
mypy == 0.790
# Tools for testing & coverage.
pytest == 5.4.1
pytest-cov == 2.10.0
pytest-mpi == 0.4
pytest-timeout == 1.4.2
mpi4py == 3.0.3
# Library dependencies.
torchtext == 0.6.0
torch >= 1.5.1
torchvision >= 0.6.0
timm == 0.3.4