Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-06-02 06:06:39 +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
mmv/models/normalization.py
+1 -1
View File
@@ -99,7 +99,7 @@ class _LayerNorm(hk.LayerNorm):
create_offset=create_offset, create_offset=create_offset,
**kwargs) **kwargs)
def __call__(self, def __call__(self, # pytype: disable=signature-mismatch # overriding-parameter-count-checks
x: types.TensorLike, x: types.TensorLike,
is_training: bool) -> jnp.ndarray: is_training: bool) -> jnp.ndarray:
del is_training # Unused. del is_training # Unused.
perceiver/perceiver.py
+1 -1
View File
@@ -682,7 +682,7 @@ class MultimodalDecoder(AbstractPerceiverDecoder):
position_encoding_type='none', position_encoding_type='none',
**decoder_kwargs) **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): subsampled_points=None):
# Partition the flat inputs among the different modalities # Partition the flat inputs among the different modalities
inputs = io_processors.restructure(modality_sizes, inputs) inputs = io_processors.restructure(modality_sizes, inputs)