torch==1.9.1
numpy==1.21.2
pytest==6.2.2