Update tests to run Python 3.9, PyTorch 1.10, torchvision 0.11.1

pull/989/head
Ross Wightman 3 years ago
parent ee40b582bb
commit 9d6aad44f8

@ -16,9 +16,9 @@ jobs:
strategy: strategy:
matrix: matrix:
os: [ubuntu-latest, macOS-latest] os: [ubuntu-latest, macOS-latest]
python: ['3.8'] python: ['3.9']
torch: ['1.9.0'] torch: ['1.10.0']
torchvision: ['0.10.0'] torchvision: ['0.11.1']
runs-on: ${{ matrix.os }} runs-on: ${{ matrix.os }}
steps: steps:
@ -30,7 +30,7 @@ jobs:
- name: Install testing dependencies - name: Install testing dependencies
run: | run: |
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install pytest pytest-timeout pip install pytest pytest-timeout expecttest
- name: Install torch on mac - name: Install torch on mac
if: startsWith(matrix.os, 'macOS') if: startsWith(matrix.os, 'macOS')
run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }} run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }}

Loading…
Cancel
Save