Run unit tests against pytorch 2.0

Balandat · Jan 29, 2025 · 2789d18 · 2789d18
1 parent 3763b6e
commit 2789d18
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/run_test_suite.yml b/.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

diff --git a/.github/workflows/run_type_checked_test_suite.yml b/.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