From 4d282bd7c1d135d65746e992bf8a1bb602759221 Mon Sep 17 00:00:00 2001 From: Lizhi Zhou <1432185+reasonsolo@users.noreply.github.com> Date: Thu, 22 Jan 2026 13:08:14 +0800 Subject: [PATCH] [https://nvbugs/5821433][fix] fix test_auto_scaling for 2 GPUs (#10866) Signed-off-by: Lizhi Zhou <1432185+reasonsolo@users.noreply.github.com> Signed-off-by: Wangshanshan <30051912+dominicshanshan@users.noreply.github.com> --- tests/integration/defs/disaggregated/test_auto_scaling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/defs/disaggregated/test_auto_scaling.py b/tests/integration/defs/disaggregated/test_auto_scaling.py index 1dd37a17a1..0f31ce54c7 100644 --- a/tests/integration/defs/disaggregated/test_auto_scaling.py +++ b/tests/integration/defs/disaggregated/test_auto_scaling.py @@ -465,7 +465,7 @@ async def test_worker_restart(model_name, disagg_server_config, worker_config, worker_config, work_dir, port=0, - device=2) + device=0) await wait_for_disagg_server_status(disagg_port, 1, 1) await asyncio.sleep(CHECK_STATUS_INTERVAL) verify_cluster_info(True, 1, 1, port=disagg_port) @@ -487,7 +487,7 @@ async def test_worker_restart(model_name, disagg_server_config, worker_config, worker_config, work_dir, port=0, - device=3) + device=1) await wait_for_disagg_server_status(disagg_port, 1, 1) await asyncio.sleep(CHECK_STATUS_INTERVAL) verify_cluster_info(True, 1, 1, port=disagg_port)