Need to specify types for collection assignment

Fixes:

fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py:66: error: Unsupported target for indexed assignment ("Mapping[str, Any]")  [index]
fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py:67: error: Unsupported target for indexed assignment ("Mapping[str, Any]")  [index]
fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py:75: error: Unsupported target for indexed assignment ("Mapping[str, Any]")  [index]
fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py:82: error: Unsupported target for indexed assignment ("Mapping[str, Any]")  [index]
fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py:87: error: Unsupported target for indexed assignment ("Mapping[str, Any]")  [index]

fms_mo/quant/quantizers.py:4161: error: Incompatible types in assignment (expression has type "list[int]", variable has type "None")  [assignment]
fms_mo/quant/quantizers.py:4161: error: Argument 1 to "len" has incompatible type "None"; expected "Sized"  [arg-type]
fms_mo/quant/quantizers.py:4162: error: Unsupported target for indexed assignment ("None")  [index]
fms_mo/quant/quantizers.py:4162: error: Value of type "None" is not indexable  [index]

Signed-off-by: Martin Hickey <[email protected]>

Author: hickeyma

fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py

@@ -14,7 +14,7 @@
 """Implement FMS adapter for INT8xINT8 checkpoints"""
 
 # Standard
-from typing import Mapping
+from typing import Mapping, MutableMapping
 
 # Third Party
 from fms.utils import serialization
@@ -46,7 +46,7 @@ def _int8_qparams_aiu(
 
 
 def _add_defaults_and_concat(
-    new_sd: Mapping[str, torch.Tensor],
+    new_sd: MutableMapping[str, torch.Tensor],
     modules_seen: set,
 ) -> None:
     """

fms_mo/quant/quantizers.py

@@ -4035,8 +4035,8 @@ def __init__(
         self.reset_ReSig_param(multimodal)
 
         self.beta = 2 / 3
-        self.Wshape = None
-        self.reshape2 = None
+        self.Wshape: list[Any] = list()
+        self.reshape2: list[Any] = list()
 
     def forward(self, x):
         if self.useSAWB: