Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-02-05 19:26:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Silence some pytype errors.

Browse Source 
PiperOrigin-RevId: 527929776
This commit is contained in:
Rebecca Chen
2023-04-28 18:53:17 +01:00
committed by Saran Tunyasuvunakool
parent fb1d757863
commit 9176a9f23c
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mmv/models/normalization.py
View File

@@ -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.

2
perceiver/perceiver.py
View File

@@ -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)