* Removed Pytorch 1.8 tests

* removed unnecessary if statement

Co-authored-by: Kunal Chakrabarty <kchakrabarty@learnfair1865.h2.fair>