Run unit tests against pytorch 2.0

Author: gpleiss

.github/workflows/run_test_suite.yml

@@ -23,7 +23,8 @@ jobs:
         if [[ ${{ matrix.pytorch-version }} = "latest" ]]; then
           pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html;
         else
-          pip install torch==1.11+cpu -f https://download.pytorch.org/whl/torch_stable.html;
+          pip install "numpy<2"  # Numpy 2.0 is not fully supported until PyTorch 2.2
+          pip install torch==2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
         fi
         pip install -e ".[test]"
     - name: Run unit tests

.github/workflows/run_type_checked_test_suite.yml

@@ -27,8 +27,8 @@ jobs:
         if [[ ${{ matrix.pytorch-version }} = "latest" ]]; then
           pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
         else
+          pip install "numpy<2"  # Numpy 2.0 is not fully supported until PyTorch 2.2
           pip install torch==2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
-          pip install "numpy<2"
         fi
         pip install -e ".[test]"
     - name: Run unit tests