From bae9e8738c99968c87f85b75fb1e117006994f5b Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Fri, 22 Nov 2024 11:42:01 -0500 Subject: [PATCH] enable test filtering w/ --providers inference=... --- .../providers/tests/inference/conftest.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/llama_stack/providers/tests/inference/conftest.py b/llama_stack/providers/tests/inference/conftest.py index d013d6a9e..7fe19b403 100644 --- a/llama_stack/providers/tests/inference/conftest.py +++ b/llama_stack/providers/tests/inference/conftest.py @@ -6,6 +6,8 @@ import pytest +from ..conftest import get_provider_fixture_overrides + from .fixtures import INFERENCE_FIXTURES @@ -67,11 +69,12 @@ def pytest_generate_tests(metafunc): indirect=True, ) if "inference_stack" in metafunc.fixturenames: - metafunc.parametrize( - "inference_stack", - [ - pytest.param(fixture_name, marks=getattr(pytest.mark, fixture_name)) - for fixture_name in INFERENCE_FIXTURES - ], - indirect=True, - ) + fixtures = INFERENCE_FIXTURES + if filtered_stacks := get_provider_fixture_overrides( + metafunc.config, + { + "inference": INFERENCE_FIXTURES, + }, + ): + fixtures = [stack.values[0]["inference"] for stack in filtered_stacks] + metafunc.parametrize("inference_stack", fixtures, indirect=True)