allowing numpy 2 in tests (openai#2362)

* allowing numpy 2 in tests * allowing numpy 2 in tests
bphd · Sep 30, 2024 · 260bbcf · 260bbcf
1 parent 25e5c36
commit 260bbcf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -44,20 +44,26 @@ jobs:
         include:
           - python-version: '3.8'
             pytorch-version: 1.13.1
+            numpy-requirement: "'numpy<2'"
           - python-version: '3.8'
             pytorch-version: 2.0.1
+            numpy-requirement: "'numpy<2'"
           - python-version: '3.9'
             pytorch-version: 2.1.2
+            numpy-requirement: "'numpy<2'"
           - python-version: '3.10'
             pytorch-version: 2.2.2
+            numpy-requirement: "'numpy<2'"
           - python-version: '3.11'
             pytorch-version: 2.3.1
+            numpy-requirement: "'numpy'"
           - python-version: '3.12'
             pytorch-version: 2.4.1
+            numpy-requirement: "'numpy'"
     steps:
       - uses: conda-incubator/setup-miniconda@v2
       - run: conda install -n test ffmpeg python=${{ matrix.python-version }}
       - uses: actions/checkout@v3
       - run: echo "$CONDA/envs/test/bin" >> $GITHUB_PATH
-      - run: pip3 install .["dev"] 'numpy<2' torch==${{ matrix.pytorch-version }}+cpu --index-url https://download.pytorch.org/whl/cpu --extra-index-url https://pypi.org/simple
+      - run: pip3 install .["dev"] ${{ matrix.numpy-requirement }} torch==${{ matrix.pytorch-version }}+cpu --index-url https://download.pytorch.org/whl/cpu --extra-index-url https://pypi.org/simple
       - run: pytest --durations=0 -vv -k 'not test_transcribe or test_transcribe[tiny] or test_transcribe[tiny.en]' -m 'not requires_cuda'