phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-31 16:01:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

dont add together fixture

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-07 21:07:38 -08:00
parent 49b68942e8
commit 0297111dfd
1 changed files with 3 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

22
llama_stack/providers/tests/safety/fixtures.py
View file

@ -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