CI Downgrade numpy to <2.0 for Mac and Windows (#1871)

huggingface · Jun 18, 2024 · 5364351 · 5364351
1 parent 717db6e
commit 5364351
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -48,6 +48,12 @@ jobs:
  python -m pip install --upgrade pip
  # cpu version of pytorch
  pip install -e .[test]
+ - name: Downgrade numpy on MacOS and Windows
+ # TODO: remove numpy downgrade on MacOS & Windows once torch fixes numpy 2.0 issue
+ shell: bash
+ if: matrix.os == 'windows-latest' || matrix.os == 'macos-12'
+ run: |
+ pip install --force-reinstall -U "numpy<2.0.0"
  - name: Test with pytest
  run: |
  make test