Testing matrix pytorch version

Author: clee-ai

.github/workflows/tests.yaml

@@ -12,6 +12,7 @@ jobs:
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
         python-version: [3.7, 3.8]
+        torch-version: [1.10.0, 1.11.0, 1.12.0]
     runs-on: ${{ matrix.os }}
     steps:
         - uses: actions/checkout@v2
@@ -24,12 +25,12 @@ jobs:
               python -m pip install --upgrade pip
               pip install "numpy<1.20" scikit-learn flake8 setuptools numba==0.49.1
 
-        - name: Install torch windows + linux
+        - name: Install torch ${{ matrix.torch-version }} windows + linux
           if: ${{matrix.os != 'macos-latest'}}
-          run: pip install torch==1.8.0+cpu torchvision==0.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
-        - name: Install torch macos
+          run: pip install torch==${{ matrix.torch-version }}+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        - name: Install torch ${{ matrix.torch-version }} macos
           if: ${{matrix.os == 'macos-latest'}}
-          run: pip install torch==1.8.0
+          run: pip install torch==${{ matrix.torch-version }}
 
         - name: Build package
           run: |