From 222e00fc4450ee1fc2fa7cd408a486a05f7243ab Mon Sep 17 00:00:00 2001 From: Stefano Castagnetta Date: Wed, 14 Oct 2026 12:17:59 -0700 Subject: [PATCH] [CI/Build] Fix test_deepseek_v41_mxfp8_scale_loading after #57797 (#61678) Signed-off-by: Stefano Castagnetta Co-authored-by: Misha Goin --- tests/quantization/test_fp8.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/quantization/test_fp8.py b/tests/quantization/test_fp8.py index 64341843d9..ae98ea6360 100644 --- a/tests/quantization/test_fp8.py +++ b/tests/quantization/test_fp8.py @@ -79,7 +79,9 @@ def test_deepseek_v41_mxfp8_scale_loading( quant_config._resolved_expert_dtype = "fp4" for module in (model_module, linear_module): monkeypatch.setattr(module, "get_tensor_model_parallel_world_size", lambda: 2) - monkeypatch.setattr(module, "get_tensor_model_parallel_rank", lambda: tp_rank) + monkeypatch.setattr( + linear_module, "get_tensor_model_parallel_rank", lambda: tp_rank + ) block = torch.nn.Module() block.attn = torch.nn.Module() block.attn.wq_b = ColumnParallelLinear(