[Tiny] Fix assert syntax warning in compressed_tensors_w4a4_mxint4_moe.py (#18899)

2026-02-17 12:54:30 +08:00
parent 7e41ac6c8d
commit 90a0d66e1e
1 changed files with 2 additions and 3 deletions
--- a/python/sglang/srt/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a4_mxint4_moe.py
+++ b/python/sglang/srt/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a4_mxint4_moe.py
@@ -89,9 +89,8 @@ class CompressedTensorsMxInt4MoE(CompressedTensorsMoEScheme):
        **extra_weight_attrs,
    ):
        assert (
-            params_dtype == torch.bfloat16,
-            f"Params dtype should be torch.bfloat16, but got: {params_dtype}",
-        )
+            params_dtype == torch.bfloat16
+        ), f"Params dtype should be torch.bfloat16, but got: {params_dtype}"

        extra_weight_attrs.update({"quant_method": self.strategy})
        w13_weight = torch.nn.Parameter(