torch==1.7.1
transformers==4.3.3
pytorch-lightning==1.2.1
scikit-learn==0.24.1
black==20.8b1
pre-commit==2.10.1

[test]
pytest
