Fixes a pytype issue.

PiperOrigin-RevId: 399305352
tensorflow · Sep 27, 2021 · 6cde0ce · 6cde0ce
1 parent 0dbe246
commit 6cde0ce
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]: # pytype: disable=invalid-annotation # typed-keras
+def _create_noise_inputs(num_upsampling_blocks: int) -> List[tf.Tensor]:
  """Creates the noise input layers."""
  noise_dimensions = get_noise_dimensions(num_upsampling_blocks)
  noise_inputs = [