Skip to content

Commit 5495560

Browse files
try to disable mypy err for the entire file
Signed-off-by: chichun-charlie-liu <[email protected]>
1 parent ee20c81 commit 5495560

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

fms_mo/quant/quantizers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
"""
2424

2525
# pylint: disable=too-many-return-statements
26+
# mypy: disable-error-code="assignment"
2627

2728
# Standard
2829
from collections.abc import Mapping
@@ -3932,7 +3933,7 @@ def init_quantization_scale(self, x: torch.Tensor, channel_wise: bool = False):
39323933
else:
39333934
if "max" in self.scale_method:
39343935
x_min = min(x.min().item(), 0)
3935-
x_max = max(x.max().item(), 0) # type: ignore[assignment]
3936+
x_max = max(x.max().item(), 0)
39363937
if "scale" in self.scale_method:
39373938
x_min = x_min * (self.n_bits + 2) / 8
39383939
x_max = x_max * (self.n_bits + 2) / 8

0 commit comments

Comments
 (0)