Skip to content

Commit

Permalink
Try to force numpy<2.0 for torch 1.13 tests, update newest tested tor…
Browse files Browse the repository at this point in the history
…ch to 2.5.1
  • Loading branch information
rwightman committed Jan 29, 2025
1 parent b1752ee commit cdbafd9
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ jobs:
matrix:
os: [ubuntu-latest]
python: ['3.10', '3.12']
torch: [{base: '1.13.0', vision: '0.14.0'}, {base: '2.4.1', vision: '0.19.1'}]
torch: [{base: '1.13.0', vision: '0.14.0'}, {base: '2.5.1', vision: '0.20.1'}]
testmarker: ['-k "not test_models"', '-m base', '-m cfg', '-m torchscript', '-m features', '-m fxforward', '-m fxbackward']
exclude:
- python: '3.12'
Expand Down Expand Up @@ -50,6 +50,9 @@ jobs:
- name: Install requirements
run: |
pip install -r requirements.txt
- name: Force old numpy for old torch
if: ${{ matrix.torch.base == '1.13.0' }}
run: pip install --upgrade 'numpy<2.0'
- name: Run tests on Windows
if: startsWith(matrix.os, 'windows')
env:
Expand Down

0 comments on commit cdbafd9

Please sign in to comment.