Add skip statements for cases where requirements-test.txt is not inst…

…alled
nod-ai · Oct 29, 2024 · 6057753 · 6057753
1 parent 7c39517
commit 6057753
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/shortfin/tests/apps/llm/cpu_llm_server_test.py b/shortfin/tests/apps/llm/cpu_llm_server_test.py
@@ -8,6 +8,9 @@
 import tempfile
 import shutil
 
+pytest.importorskip("transformers", "transformers is not installed")
+pytest.importorskip("tokenizers", "tokenizers is not installed")
+
 BATCH_SIZES = [1, 4]
 
 cpu_settings = {