Silence some pytype errors.

PiperOrigin-RevId: 527929776
google-deepmind · Jun 2, 2023 · 9176a9f · 9176a9f
1 parent fb1d757
commit 9176a9f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mmv/models/normalization.py b/mmv/models/normalization.py
@@ -99,7 +99,7 @@ def __init__(self,
                      create_offset=create_offset,
                      **kwargs)
 
-  def __call__(self,
+  def __call__(self,  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
                x: types.TensorLike,
                is_training: bool) -> jnp.ndarray:
     del is_training  # Unused.

diff --git a/perceiver/perceiver.py b/perceiver/perceiver.py
@@ -682,7 +682,7 @@ def __init__(self, modalities, num_outputs, output_num_channels,
         position_encoding_type='none',
         **decoder_kwargs)
 
-  def decoder_query(self, inputs, modality_sizes, inputs_without_pos=None,
+  def decoder_query(self, inputs, modality_sizes, inputs_without_pos=None,  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
                     subsampled_points=None):
     # Partition the flat inputs among the different modalities
     inputs = io_processors.restructure(modality_sizes, inputs)