From 23210a911eb9c8bfb627d96481244ac4360877b3 Mon Sep 17 00:00:00 2001
From: Cyrus Leung <tlleungac@connect.ust.hk>
Date: Thu, 19 Feb 2026 19:16:58 +0800
Subject: [PATCH] [CI/Build] Try to make beam search test less flaky (#34885)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
---
 tests/samplers/test_beam_search.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/samplers/test_beam_search.py b/tests/samplers/test_beam_search.py
index aef7eec09..98675856a 100644
--- a/tests/samplers/test_beam_search.py
+++ b/tests/samplers/test_beam_search.py
@@ -27,7 +27,7 @@ EXTRA_ENGINE_KWARGS: dict = (
         max_num_seqs=1,
     )
     if current_platform.is_rocm()
-    else {}
+    else dict(async_scheduling=False, max_num_seqs=1)
 )
 
 # FIXME(zhuohan): The test can not pass if we: