From 9176a9f23ced8e3d6024718d757be8d67cfb6927 Mon Sep 17 00:00:00 2001 From: Rebecca Chen Date: Fri, 28 Apr 2023 18:53:17 +0100 Subject: [PATCH] Silence some pytype errors. PiperOrigin-RevId: 527929776 --- mmv/models/normalization.py | 2 +- perceiver/perceiver.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mmv/models/normalization.py b/mmv/models/normalization.py index 639833f..ff186e4 100644 --- a/mmv/models/normalization.py +++ b/mmv/models/normalization.py @@ -99,7 +99,7 @@ class _LayerNorm(hk.LayerNorm): 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 index 391a262..2b9dfb7 100644 --- a/perceiver/perceiver.py +++ b/perceiver/perceiver.py @@ -682,7 +682,7 @@ class MultimodalDecoder(AbstractPerceiverDecoder): 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)