phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 23:03:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use env vars for bedrock guardrail vars

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-07 20:15:42 -08:00
parent 98c09323a9
commit 932b524449
2 changed files with 3 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -71,18 +71,6 @@ def pytest_addoption(parser):
default=None,
help="Specify the safety model to use for testing",
)
parser.addoption(
"--bedrock-guardrail-id",
action="store",
default=None,
help="Specify the guard rail ID to use for testing bedrock",
)
parser.addoption(
"--bedrock-guardrail-version",
action="store",
default=None,
help="Specify the guard rail version to use for testing bedrock",
)
SAFETY_MODEL_PARAMS = [