From 19d57b4d8227ec5e11201b7ccb03d62b4ef9ff5e Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Thu, 7 Nov 2024 21:07:38 -0800 Subject: [PATCH] dont add together fixture --- .../providers/tests/safety/fixtures.py | 22 +++---------------- 1 file changed, 3 insertions(+), 19 deletions(-) diff --git a/llama_stack/providers/tests/safety/fixtures.py b/llama_stack/providers/tests/safety/fixtures.py index 40c89de92..000318d5d 100644 --- a/llama_stack/providers/tests/safety/fixtures.py +++ b/llama_stack/providers/tests/safety/fixtures.py @@ -50,25 +50,6 @@ def safety_meta_reference(safety_model) -> ProviderFixture: ) -@pytest.fixture(scope="session") -def safety_together() -> ProviderFixture: - return ProviderFixture( - providers=[ - Provider( - provider_id="together", - provider_type="remote::together", - config=TogetherSafetyConfig().model_dump(), - ) - ], - provider_data=dict( - together_api_key=get_env_or_fail("TOGETHER_API_KEY"), - ), - ) - - -SAFETY_FIXTURES = ["meta_reference", "together", "remote", "bedrock"] - - @pytest.fixture(scope="session") def safety_bedrock() -> ProviderFixture: return ProviderFixture( @@ -82,6 +63,9 @@ def safety_bedrock() -> ProviderFixture: ) +SAFETY_FIXTURES = ["meta_reference", "bedrock", "remote"] + + @pytest_asyncio.fixture(scope="session") async def safety_stack(inference_model, safety_model, request): # We need an inference + safety fixture to test safety