Run unit test with Pytorch CPU instead of GPU

This is so tests can be run without need for a GPU

Signed-off-by: Martin Hickey <[email protected]>

Author: hickeyma

.github/workflows/test.yml

@@ -35,8 +35,6 @@ permissions:
 
 jobs:
   test:
-    # Disable until runner with Nvidia GPU available
-    if: false
     name: "test: ${{ matrix.python }} on ${{ matrix.platform }}"
     runs-on: "${{ matrix.platform }}"
     strategy:

tox.ini

@@ -4,6 +4,9 @@ minversion = 4.4
 
 [testenv]
 description = run tests (unit)
+# Use PyTorch CPU build instead of CUDA build in test envs
+setenv =
+    PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cpu
 extras =
     dev
 package = wheel