[CI][Bugfix] Fix MoE reprocess test mock after #57405's kernel refactor (#57641)

Signed-off-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Signed-off-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
Co-authored-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
This commit is contained in:
vllm-agent
2026-09-19 21:42:21 +00:00
committed by GitHub
co-authored by kimi-no-na-wa Wentao Ye
parent 36fa72d2d0
commit 674b6d95d6
+3 -1
View File
@@ -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()