phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 18:12:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added nvidia as safety provider

Browse source
This commit is contained in:
Chantal D Gama Rose 2025-02-25 08:16:49 +00:00
parent 07a992ef90
commit 0593408c19
14 changed files with 354 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/tests/safety/conftest.py
View file

@ -51,11 +51,19 @@ DEFAULT_PROVIDER_COMBINATIONS = [
id="remote",
marks=pytest.mark.remote,
),
pytest.param(
{
"inference": "nvidia",
"safety": "nvidia",
},
id="nvidia",
marks=pytest.mark.nvidia,
),
]
def pytest_configure(config):
for mark in ["meta_reference", "ollama", "together", "remote", "bedrock"]:
for mark in ["meta_reference", "ollama", "together", "remote", "bedrock", "nvidia"]:
config.addinivalue_line(
"markers",
f"{mark}: marks tests as {mark} specific",