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