From 684de1e6fdeea5cbdbede222308b998ab4878ecd Mon Sep 17 00:00:00 2001 From: Sixian Yi Date: Tue, 21 Jan 2025 17:32:34 -0800 Subject: [PATCH] report --- tests/client-sdk/conftest.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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"