Ignore some pytype errors.

PiperOrigin-RevId: 871279957

Author: QEDady

scenic/projects/avatar/models.py

@@ -655,7 +655,7 @@ def get_metrics_fn(self, split: Optional[str] = None):
     """
     del split  # The metric function is the same for all splits.
 
-    def metric_fn(
+    def metric_fn(  # pytype: disable=annotation-type-mismatch
         logits: jnp.ndarray,
         targets: jnp.ndarray,
         weights: jnp.ndarray,

scenic/projects/pointcloud/segmentation_model.py

@@ -62,7 +62,7 @@ def point_count(logits: jnp.ndarray,
 })
 
 
-def semantic_segmentation_metrics_function(
+def semantic_segmentation_metrics_function(  # pytype: disable=annotation-type-mismatch
     logits: jnp.ndarray,
     batch: base_model.Batch,
     target_is_onehot: bool = False,