Caching MNIST dataset for testing (#917)

* Caching MNIST dataset for testing

* Added MNIST datset to the tests directory

* Caches dataset based off hash of the test.pt file

* Cleaned Up yml file

* Cleaned Up yml file

* Removed MNIST Data from framework

* Set cache key for dataset to 'mnist'

* Apply suggestions from code review

* Apply suggestions from code review

Co-authored-by: Jirka Borovec <[email protected]>

Author: djbyrne

.github/workflows/ci-testing.yml

@@ -37,6 +37,13 @@ jobs:
           ${{ runner.os }}-pip-${{ matrix.python-version }}-
           ${{ runner.os }}-${{ matrix.python-version }}-
 
+    - name: Cache datasets
+      uses: actions/cache@v1
+      with:
+        path: tests/models/mnist # This path is specific to Ubuntu
+        # Look to see if there is a cache hit for the corresponding requirements file
+        key: mnist-dataset
+
     - name: Install dependencies
       run: |
         # python -m pip install --upgrade --user pip