pytest==7.*
torch==2.*
