From 674b6d95d6d0bd63b44cae0abe2bf304328c73e9 Mon Sep 17 00:00:00 2001 From: vllm-agent Date: Sat, 19 Sep 2026 14:42:21 -0700 Subject: [PATCH] [CI][Bugfix] Fix MoE reprocess test mock after #57405's kernel refactor (#57641) Signed-off-by: kimi-no-na-wa Signed-off-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com> Co-authored-by: kimi-no-na-wa Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com> --- tests/kernels/moe/test_flashinfer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/kernels/moe/test_flashinfer.py b/tests/kernels/moe/test_flashinfer.py index d17416de99..9189c5c641 100644 --- a/tests/kernels/moe/test_flashinfer.py +++ b/tests/kernels/moe/test_flashinfer.py @@ -2,6 +2,7 @@ # SPDX-FileCopyrightText: Copyright contributors to the vLLM project import multiprocessing from dataclasses import dataclass +from unittest.mock import MagicMock import pytest import torch @@ -537,10 +538,11 @@ def _make_unquantized_flashinfer_test_layer( method.moe = moe_config method.unquantized_backend = UnquantizedMoeBackend.FLASHINFER_TRTLLM method.moe_kernel = None + mock_kernel = MagicMock() monkeypatch.setattr( method, "_init_moe_kernel", - lambda _: setattr(method, "moe_kernel", object()), + lambda _: setattr(method, "moe_kernel", mock_kernel), ) layer = torch.nn.Module()