From 45f438c027176c783a1068894b9147eb008ffa5e Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Mon, 29 Sep 2025 16:11:37 -0400 Subject: [PATCH] chore: skip safety tests when shield not available (#3592) # What does this PR do? we skip embedding tests when the embedding_model_id isn't provided. same for completion / chat tests when text_model_id isn't given. instead of failing safety tests when a shield_id isn't provided, we'll skip them too. ## Test Plan ci Co-authored-by: Ashwin Bharambe --- tests/integration/fixtures/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/fixtures/common.py b/tests/integration/fixtures/common.py index ee4c5755a..835a701bc 100644 --- a/tests/integration/fixtures/common.py +++ b/tests/integration/fixtures/common.py @@ -166,7 +166,7 @@ def model_providers(llama_stack_client): @pytest.fixture(autouse=True) def skip_if_no_model(request): - model_fixtures = ["text_model_id", "vision_model_id", "embedding_model_id", "judge_model_id"] + model_fixtures = ["text_model_id", "vision_model_id", "embedding_model_id", "judge_model_id", "shield_id"] test_func = request.node.function actual_params = inspect.signature(test_func).parameters.keys()