From f68af3098cae9d302b1d45804ad66f3632b35f75 Mon Sep 17 00:00:00 2001
From: Jong Wook Kim <jongwook@nyu.edu>
Date: Mon, 30 Sep 2024 11:11:54 -0700
Subject: [PATCH] allowing numpy 2 in tests

---
 .github/workflows/test.yml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index a1cc48d..1d002e0 100644
--- 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'