Silence some pytype errors.

PiperOrigin-RevId: 398607556
tensorflow · Sep 23, 2021 · 0dbe246 · 0dbe246
1 parent 907171a
commit 0dbe246
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_graphics/projects/gan/architectures_style_gan_v2.py b/tensorflow_graphics/projects/gan/architectures_style_gan_v2.py
@@ -85,7 +85,7 @@ def get_noise_dimensions(
  return noise_dimensions
 
 
-def _create_noise_inputs(num_upsampling_blocks: int) -> List[tf.keras.Input]:
+def _create_noise_inputs(num_upsampling_blocks: int) -> List[tf.keras.Input]: # pytype: disable=invalid-annotation # typed-keras
  """Creates the noise input layers."""
  noise_dimensions = get_noise_dimensions(num_upsampling_blocks)
  noise_inputs = [