fix test

Adel-Moumen · Adel-Moumen · commit 60f2770f7d8e · 2022-12-06T20:29:58.000+01:00
diff --git a/conftest.py b/conftest.py
@@ -23,6 +23,10 @@ def pytest_generate_tests(metafunc):
     from transformers import Wav2Vec2Model  # noqa: F401
 except ModuleNotFoundError:
     collect_ignore.append("speechbrain/lobes/models/huggingface_wav2vec.py")
+try:
+    from transformers import WhisperModel  # noqa: F401
+except ModuleNotFoundError:
+    collect_ignore.append("speechbrain/lobes/models/huggingface_whisper.py")
 try:
     import sacrebleu  # noqa: F401
 except ModuleNotFoundError:
