phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-18 07:39:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix eval tests model registration

Browse source
This commit is contained in:
Xi Yan 2024-12-11 10:36:39 -08:00
parent 3b5a33d921
commit 00658e02f8
3 changed files with 26 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/tests/eval/conftest.py
View file

@ -80,6 +80,13 @@ def pytest_addoption(parser):
help="Specify the inference model to use for testing",
)
parser.addoption(
"--judge-model",
action="store",
default="meta-llama/Llama-3.1-8B-Instruct",
help="Specify the judge model to use for testing",
)
def pytest_generate_tests(metafunc):
if "eval_stack" in metafunc.fixturenames: