quick fix

IwakuraRein · IwakuraRein · commit 33acaa241943 · 2026-02-13T21:05:05.000Z
Signed-off-by: Siyuan Fu &lt;siyuanf@nvidia.com&gt;
diff --git a/tests/moe/test_dpsk_fused_moe_fp8.py b/tests/moe/test_dpsk_fused_moe_fp8.py
@@ -597,7 +597,7 @@ def test_correctness_dpsk_fp8_fused_moe(
     class FP8BlockScaleMoe:
         def __init__(self):
             self.name = "FP8BlockScale"
-            self.quant_mode = QuantMode.FP8_BLOCK_SCALE
+            self.quant_mode = QuantMode.FP8_BLOCK_SCALE_DEEPSEEK
 
     moe_impl = FP8BlockScaleMoe()
 
