Update skip on torch in tests (#3136)

Commit

2 years ago

Update skip on torch in tests (#3136) * Replace old torch version checks with existing function * Clean up formatting