From ea0fa34f4992c09fc5aa4ae9a12e67e03125289c Mon Sep 17 00:00:00 2001 From: Lucas Wilkinson Date: Thu, 9 Jul 2026 16:19:03 -0400 Subject: [PATCH] [CI] Increase extract hidden states TP2 timeout (#48161) Signed-off-by: Lucas Wilkinson --- .../extract_hidden_states_integration/test_extraction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py b/tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py index 4a84ed0314..82e0eb9d8f 100644 --- a/tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py +++ b/tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py @@ -297,7 +297,7 @@ def test_extract_hidden_states_qwen35_hybrid_smoke(tmp_path): ) -@pytest.mark.timeout(240 if current_platform.is_rocm() else 60) +@pytest.mark.timeout(240 if current_platform.is_rocm() else 120) @multi_gpu_test(num_gpus=2) @create_new_process_for_each_test() def test_extract_hidden_states_tp2():