Fixing tests (#936)

* abs import

* rename test model

* update trainer

* revert test_step check

* move tags

* fix test_step

* clean tests

* fix template

* update dataset path

* fix parent order

.github/workflows/ci-testing.yml

@@ -40,7 +40,7 @@ jobs:
     - name: Cache datasets
       uses: actions/cache@v1
       with:
-        path: tests/models/mnist # This path is specific to Ubuntu
+        path: tests/datasets # This path is specific to Ubuntu
         # Look to see if there is a cache hit for the corresponding requirements file
         key: mnist-dataset