Skip to content

Commit 7ae5917

Browse files
Update tests.yml
1 parent 40171b5 commit 7ae5917

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

.github/workflows/tests.yml

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,8 @@ jobs:
101101
fail-fast: false
102102
matrix:
103103
os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15]
104-
# Test with the oldest supported torch version, the newest stable, current RC, and nightly.
105-
torch_version: ["2.2.2", "2.7.1", "2.8.0", "2.9.0"]
104+
# Test with the oldest supported torch version, the newest two stable/RC.
105+
torch_version: ["2.2.2", "2.7.1", "2.8.0"]
106106
include:
107107
- os: ubuntu-22.04
108108
arch: x86_64
@@ -144,7 +144,7 @@ jobs:
144144

145145
- name: Install dependencies
146146
run: |
147-
pip install torch~=${{ matrix.torch_version }}.dev0 --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || (matrix.torch_version == '2.9.0' && 'nightly/cpu') || 'cpu' }}
147+
pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || 'cpu' }}
148148
pip install -e ".[test]"
149149
pip install pytest-cov
150150
@@ -373,9 +373,6 @@ jobs:
373373
- cuda_version: "12.9.1"
374374
torch_version: "2.8.0"
375375
pypi_index: "https://download.pytorch.org/whl/test/cu129"
376-
- cuda_version: "12.9.1"
377-
torch_version: "2.9.0"
378-
pypi_index: "https://download.pytorch.org/whl/nightly/cu129"
379376

380377

381378
# Linux L40S runners

0 commit comments

Comments
 (0)