pytest==7.1.2
torch==1.13
