From 60577532acf451371adfcf75eac2ed78cb1ed9ce Mon Sep 17 00:00:00 2001 From: Stephen Baione Date: Tue, 29 Oct 2024 16:25:26 -0500 Subject: [PATCH] Add skip statements for cases where requirements-test.txt is not installed --- shortfin/tests/apps/llm/cpu_llm_server_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/shortfin/tests/apps/llm/cpu_llm_server_test.py b/shortfin/tests/apps/llm/cpu_llm_server_test.py index c69ad41b6..4fd7066de 100644 --- 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 = {