Remove flair and spacy tests as they are incompatible with numpy 2.0

pplantinga · pplantinga · commit 6d8167dee511 · 2025-03-23T17:16:33.000-04:00
diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
@@ -30,7 +30,7 @@ jobs:
               run: |
                   pip install uv
                   uv pip install --system ctc-segmentation sacrebleu # ctc-segmentation is funky with uv due to their oldest-supported-numpy dependency
-                  uv pip install --system -r requirements.txt torch==2.6.0+cpu torchaudio==2.6.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu k2==1.24.4.dev20250307+cpu.torch2.6.0 --find-links https://k2-fsa.github.io/k2/cpu.html kaldilm==1.15.1 spacy==3.7.4 flair==0.13.1 gensim==4.3.2
+                  uv pip install --system -r requirements.txt torch==2.6.0+cpu torchaudio==2.6.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu k2==1.24.4.dev20250307+cpu.torch2.6.0 --find-links https://k2-fsa.github.io/k2/cpu.html kaldilm==1.15.1 gensim==4.3.2
                   uv pip install --system --editable . --no-deps  # already installed pinned deps from requirements.txt, we're good
             - name: Install sox
               run: |
diff --git a/conftest.py b/conftest.py
@@ -13,6 +13,8 @@ def pytest_generate_tests(metafunc):
 collect_ignore = [
     "setup.py",
     "speechbrain/lobes/models/huggingface_transformers/mert.py",
+    "speechbrain/lobes/models/flair",
+    "speechbrain/lobes/models/spacy",
 ]
 try:
     import numba  # noqa: F401
diff --git a/tests/.run-doctests.sh b/tests/.run-doctests.sh
@@ -5,5 +5,5 @@ set -e -u -o pipefail
 # > pytest --doctest-modules speechbrain/
 # However, we take this more complex approach to avoid testing files not
 # tracked by git. We filter out tests that require optional dependencies.
-avoid="transducer_loss.py\|fairseq_wav2vec.py\|huggingface_wav2vec.py\|bleu.py\|ctc_segmentation.py\|check_url.py\|huggingface_whisper.py\|language_model.py\|vocos.py\|discrete_ssl.py\|lobes/flair|lobes/spacy"
+avoid="transducer_loss.py\|fairseq_wav2vec.py\|huggingface_wav2vec.py\|bleu.py\|ctc_segmentation.py\|check_url.py\|huggingface_whisper.py\|language_model.py\|vocos.py\|discrete_ssl.py\|lobes/models/flair\|lobes/models/spacy"
 git ls-files speechbrain | grep -e "\.py$" | grep -v $avoid | xargs pytest --doctest-modules

Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,8 @@ def pytest_generate_tests(metafunc):`
`13`	`13`	`collect_ignore = [`
`14`	`14`	`"setup.py",`
`15`	`15`	`"speechbrain/lobes/models/huggingface_transformers/mert.py",`
	`16`	`+ "speechbrain/lobes/models/flair",`
	`17`	`+ "speechbrain/lobes/models/spacy",`
`16`	`18`	`]`
`17`	`19`	`try:`
`18`	`20`	`import numba # noqa: F401`