From 9d6aad44f8fd32e89e5cca503efe3ada5071cc2a Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 18 Nov 2021 21:38:04 -0800
Subject: [PATCH] Update tests to run Python 3.9, PyTorch 1.10, torchvision
 0.11.1

---
 .github/workflows/tests.yml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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