mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-02 17:54:31 +00:00
[ez] move addoption to conftest in root test directory and fix broken imports
This commit is contained in:
parent
965644ce68
commit
4d3f0ae79b
9 changed files with 28 additions and 81 deletions
|
|
@ -76,22 +76,6 @@ def pytest_configure(config):
|
|||
)
|
||||
|
||||
|
||||
def pytest_addoption(parser):
|
||||
parser.addoption(
|
||||
"--inference-model",
|
||||
action="store",
|
||||
default="meta-llama/Llama-3.2-3B-Instruct",
|
||||
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:
|
||||
available_fixtures = {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue