Suppress new Pyrefly findings before upgrading to v1.2.0

PiperOrigin-RevId: 979359023

Author: oprypin

official/nlp/modeling/networks/encoder_scaffold.py

@@ -134,8 +134,8 @@ def __init__(self,
             **embedding_cfg) if embedding_cfg else embedding_cls()
       else:
         embedding_network = embedding_cls
-      inputs = embedding_network.inputs
-      embeddings, attention_mask = embedding_network(inputs)
+      inputs = embedding_network.inputs  # pyrefly: ignore[missing-attribute]
+      embeddings, attention_mask = embedding_network(inputs)  # pyrefly: ignore[not-callable]
       embedding_layer = None
       position_embedding_layer = None
       type_embedding_layer = None
@@ -193,7 +193,7 @@ def __init__(self,
         mask_layer = mask_cls(**mask_cfg)
       else:
         mask_layer = mask_cls
-      attention_mask = mask_layer(embeddings, mask)
+      attention_mask = mask_layer(embeddings, mask)  # pyrefly: ignore[not-callable]
 
     data = embeddings
 
@@ -225,9 +225,9 @@ def __init__(self,
       else:
         layer = cur_hidden_cls
       if recursive:
-        data, recursive_states = layer([data, attention_mask, recursive_states])
+        data, recursive_states = layer([data, attention_mask, recursive_states])  # pyrefly: ignore[not-callable]
       else:
-        data = layer([data, attention_mask])
+        data = layer([data, attention_mask])  # pyrefly: ignore[not-callable]
       layer_output_data.append(data)
       hidden_layers.append(layer)
 

official/projects/assemblenet/modeling/assemblenet.py

@@ -1033,7 +1033,7 @@ def build_assemblenet_v1(
       backbone_cfg.blocks)
   params = ASSEMBLENET_SPECS[assemblenet_depth]
   block_fn = functools.partial(
-      params['block'],  # pyrefly: ignore[bad-argument-type]
+      params['block'],  # pyrefly: ignore[bad-argument-type, not-callable]
       use_sync_bn=norm_activation_config.use_sync_bn,
       bn_decay=norm_activation_config.norm_momentum,
       bn_epsilon=norm_activation_config.norm_epsilon)

official/projects/edgetpu/vision/modeling/custom_layers.py

@@ -127,7 +127,7 @@ def __init__(self,
     if self.use_batch_norm:
       self.batch_norm_layer = [
           batch_norm_layer(  # pyrefly: ignore[not-callable]
-              axis=-1, momentum=self.bn_momentum, epsilon=self.bn_epsilon)
+              axis=-1, momentum=self.bn_momentum, epsilon=self.bn_epsilon)  # pyrefly: ignore[unexpected-keyword]
           for i in range(self._groups)
       ]
 

official/vision/serving/export_tflite_lib.py

@@ -141,7 +141,7 @@ def convert_tflite_model(
           representative_dataset,
           params=params,
           task=task,
-          calibration_steps=calibration_steps)
+          calibration_steps=calibration_steps)  # pyrefly: ignore[bad-argument-type]
       if quant_type.startswith('int8_full'):
         converter.target_spec.supported_ops = [
             tf.lite.OpsSet.TFLITE_BUILTINS_INT8
@@ -161,7 +161,7 @@ def convert_tflite_model(
             debug_dataset=functools.partial(
                 representative_dataset,
                 params=params,
-                calibration_steps=calibration_steps),
+                calibration_steps=calibration_steps),  # pyrefly: ignore[bad-argument-type]
             debug_options=debug_options)
         debugger.run()
         return debugger.get_nondebug_quantized_model()