Internal change

PiperOrigin-RevId: 458331830

official/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py

@@ -95,7 +95,7 @@ class UNet3DDecoder(tf.keras.Model):
       channel_dim = 1
 
     # Build 3D UNet.
-    inputs = self._build_input_pyramid(input_specs, model_id)
+    inputs = self._build_input_pyramid(input_specs, model_id)  # pytype: disable=wrong-arg-types  # dynamic-method-lookup
 
     # Add levels with up-convolution or up-sampling.
     x = inputs[str(model_id)]

official/vision/modeling/maskrcnn_model.py

@@ -158,7 +158,7 @@ class MaskRCNNModel(tf.keras.Model):
         matched_gt_classes=intermediate_outputs['matched_gt_classes'],
         gt_masks=gt_masks,
         training=training)
-    model_outputs.update(model_mask_outputs)
+    model_outputs.update(model_mask_outputs)  # pytype: disable=attribute-error  # dynamic-method-lookup
     return model_outputs
 
   def _get_backbone_and_decoder_features(self, images):