Silence some pytype errors.

PiperOrigin-RevId: 527708098

Author: rchen152

tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py

@@ -91,7 +91,7 @@ def _skip_compressible_weights(self, *args, **kwargs):
     self.dummy_name_to_tensor[name] = dummy_zeros
     return dummy_zeros
 
-  def build(self, input_shape):
+  def build(self, input_shape):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
     # Building nested layer via `super` must happen first
     # so that the nested layer's variables
     # are available to `init_training_weights`.
@@ -206,7 +206,7 @@ def _skip_compressible_weights(self, *args, **kwargs):
 
     return tf.zeros(shape)
 
-  def build(self, input_shape):
+  def build(self, input_shape):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
     super(_InferenceWrapper, self).build(input_shape)
 
     # Add weights needed by algorithm during inference.