pytest==7.*
torch==2.0.*
