mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-02 18:04:30 +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
|
|
@ -11,21 +11,6 @@ from ..conftest import get_provider_fixture_overrides
|
|||
from .fixtures import INFERENCE_FIXTURES
|
||||
|
||||
|
||||
def pytest_addoption(parser):
|
||||
parser.addoption(
|
||||
"--inference-model",
|
||||
action="store",
|
||||
default=None,
|
||||
help="Specify the inference model to use for testing",
|
||||
)
|
||||
parser.addoption(
|
||||
"--embedding-model",
|
||||
action="store",
|
||||
default=None,
|
||||
help="Specify the embedding model to use for testing",
|
||||
)
|
||||
|
||||
|
||||
def pytest_configure(config):
|
||||
for model in ["llama_8b", "llama_3b", "llama_vision"]:
|
||||
config.addinivalue_line(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue