[CI] Increase extract hidden states TP2 timeout (#48161)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
Lucas Wilkinson committed
Jul 9, 2026 at 16:19 UTC
ea0fa34f4992c09fc5aa4ae9a12e67e03125289c
1 file changed
+1
-1
tests/v1/kv_connector/extract_hidden_states_integration/test_extraction.py
+1
-1
@@ -297,7 +297,7 @@ def test_extract_hidden_states_qwen35_hybrid_smoke(tmp_path):
297
)
298
299
300
-@pytest.mark.timeout(240 if current_platform.is_rocm() else 60)
300
+@pytest.mark.timeout(240 if current_platform.is_rocm() else 120)
301
@multi_gpu_test(num_gpus=2)
302
@create_new_process_for_each_test()
303
def test_extract_hidden_states_tp2():