missing package

Author: 1adrianb

.github/workflows/test.yml

@@ -24,9 +24,9 @@ jobs:
           conda create -q -n test-environment python=${{ matrix.python-version }}
           source activate test-environment
           if [[ "${{ matrix.pytorch-version }}" == "nightly" ]]; then
-            conda install pytorch cpuonly -c pytorch-nightly 
+            conda install pytorch torchvision cpuonly -c pytorch-nightly 
           else
-            conda install pytorch==${{ matrix.pytorch-version }} cpuonly -c pytorch
+            conda install pytorch==${{ matrix.pytorch-version }} torchvision cpuonly -c pytorch
           fi
           conda install flake8
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi