From b2f2bd71ba6cff6d0678e2f83d45c1c08425144c Mon Sep 17 00:00:00 2001
From: vllm-agent <claw@inferact.ai>
Date: Thu, 17 Sep 2026 10:02:24 -0700
Subject: [PATCH] [CI][Bugfix] Add tp_shard_with_padding to padded MoE reload
 test mock (#57402)

Signed-off-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Co-authored-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Co-authored-by: Misha Goin <mgoin64@gmail.com>
---
 tests/model_executor/model_loader/test_reload.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/model_executor/model_loader/test_reload.py b/tests/model_executor/model_loader/test_reload.py
index 844149ce1b..fde8b86cda 100644
--- a/tests/model_executor/model_loader/test_reload.py
+++ b/tests/model_executor/model_loader/test_reload.py
@@ -489,6 +489,7 @@ def test_padded_moe_reload_releases_each_layer(
         is_act_and_mul=is_gated,
         has_bias=has_bias,
         tp_rank=tp_rank,
+        tp_shard_with_padding=False,
         moe_parallel_config=SimpleNamespace(tp_size=2),
     )
     model = torch.nn.ModuleList()