Skip to content

GitLab

Unverified Commit 84371daf authored by Nick Hill's avatar Nick Hill Committed by GitHub
Browse files

[Tests] Verify gpt_oss package is installed in harmony tests (#29336) 


Signed-off-by: default avatarNick Hill <nhill@redhat.com>
parent f32c7d6f
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
tests/entrypoints/openai/test_response_api_with_harmony.py
View file @ 84371daf
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
import importlib
import json import json
import time import time
...@@ -35,6 +35,10 @@ GET_WEATHER_SCHEMA = { ...@@ -35,6 +35,10 @@ GET_WEATHER_SCHEMA = {
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def server(): def server():
assert importlib.util.find_spec("gpt_oss") is not None, (
"Harmony tests require gpt_oss package to be installed"
)
args = ["--enforce-eager", "--tool-server", "demo", "--max_model_len", "5000"] args = ["--enforce-eager", "--tool-server", "demo", "--max_model_len", "5000"]
env_dict = dict( env_dict = dict(
VLLM_ENABLE_RESPONSES_API_STORE="1", VLLM_ENABLE_RESPONSES_API_STORE="1",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment