diff --git a/tests/client-sdk/conftest.py b/tests/client-sdk/conftest.py index c19546887..8889b6f04 100644 --- a/tests/client-sdk/conftest.py +++ b/tests/client-sdk/conftest.py @@ -29,6 +29,21 @@ def pytest_addoption(parser): ) +TEXT_MODEL = "meta-llama/Llama-3.1-8B-Instruct" +INFERENCE_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct" + + if config.getoption("--output"): + config.pluginmanager.register(Report()) + + +def pytest_addoption(parser): + parser.addoption( + "--output", + action="store_false", + help="Knob to determine if we should generate report, e.g. --output=True", + ) + + TEXT_MODEL = "meta-llama/Llama-3.1-8B-Instruct" INFERENCE_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"