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

fix failing memory tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-12 11:43:06 -08:00
parent c38d377eb7
commit ee3f0c6b55
3 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -58,10 +58,10 @@ DEFAULT_PROVIDER_COMBINATIONS = [
def pytest_addoption(parser):
parser.addoption(
"--embedding-model",
"--inference-model",
action="store",
default=None,
help="Specify the embedding model to use for testing",
help="Specify the inference model to use for testing",
)
@ -74,15 +74,15 @@ def pytest_configure(config):
def pytest_generate_tests(metafunc):
if "embedding_model" in metafunc.fixturenames:
model = metafunc.config.getoption("--embedding-model")
if "inference_model" in metafunc.fixturenames:
model = metafunc.config.getoption("--inference-model")
if not model:
raise ValueError(
"No embedding model specified. Please provide a valid embedding model."
"No inference model specified. Please provide a valid inference model."
)
params = [pytest.param(model, id="")]
metafunc.parametrize("embedding_model", params, indirect=True)
metafunc.parametrize("inference_model", params, indirect=True)
if "memory_stack" in metafunc.fixturenames:
available_fixtures = {
"inference": INFERENCE_FIXTURES,