From 08fa85a1adaeaca1403da53e9678012fbcb44cc4 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Mon, 2 Feb 2026 06:10:54 +0000 Subject: [PATCH] Update audio tensor access in offline inference tests and increase image fetch timeout for CI Signed-off-by: tzhouam --- tests/e2e/offline_inference/test_qwen2_5_omni.py | 2 +- tests/e2e/offline_inference/test_qwen3_omni.py | 2 +- tests/e2e/online_serving/test_image_gen_edit.py | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/e2e/offline_inference/test_qwen2_5_omni.py b/tests/e2e/offline_inference/test_qwen2_5_omni.py index 3537ea127..af9d793c1 100644 --- a/tests/e2e/offline_inference/test_qwen2_5_omni.py +++ b/tests/e2e/offline_inference/test_qwen2_5_omni.py @@ -88,7 +88,7 @@ def test_mixed_modalities_to_audio(omni_runner: type[OmniRunner], test_config: t assert len(audio_output.request_output) > 0 # Verify audio tensor exists and has content - audio_tensor = audio_output.request_output[0].multimodal_output["audio"] + audio_tensor = audio_output.request_output[0].outputs[0].multimodal_output["audio"] assert audio_tensor is not None assert audio_tensor.numel() > 0 diff --git a/tests/e2e/offline_inference/test_qwen3_omni.py b/tests/e2e/offline_inference/test_qwen3_omni.py index 470cdb73b..2c5c66fe3 100644 --- a/tests/e2e/offline_inference/test_qwen3_omni.py +++ b/tests/e2e/offline_inference/test_qwen3_omni.py @@ -75,6 +75,6 @@ def test_video_to_audio(omni_runner: type[OmniRunner], test_config) -> None: assert len(audio_output.request_output) > 0 # Verify audio tensor exists and has content - audio_tensor = audio_output.request_output[0].multimodal_output["audio"] + audio_tensor = audio_output.request_output[0].outputs[0].multimodal_output["audio"] assert audio_tensor is not None assert audio_tensor.numel() > 0 diff --git a/tests/e2e/online_serving/test_image_gen_edit.py b/tests/e2e/online_serving/test_image_gen_edit.py index e796fe959..8db0d50fb 100644 --- a/tests/e2e/online_serving/test_image_gen_edit.py +++ b/tests/e2e/online_serving/test_image_gen_edit.py @@ -23,6 +23,8 @@ from vllm.assets.image import ImageAsset from vllm.utils.network_utils import get_open_port os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn" +# Increase timeout for downloading assets from S3 (default 5s is too short for CI) +os.environ.setdefault("VLLM_IMAGE_FETCH_TIMEOUT", "60") models = ["Qwen/Qwen-Image-Edit-2509"] test_params = models