Silence some pytype errors.

rchen152 · Magenta Team · commit b2d180ac6156 · 2023-05-01T10:34:55.000-07:00
PiperOrigin-RevId: 528510090
diff --git a/ddsp/training/metrics.py b/ddsp/training/metrics.py
@@ -181,7 +181,7 @@ def __init__(self, sample_rate, frame_rate, name='loudness'):
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, audio_gen):
+  def update_state(self, batch, audio_gen):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
     """Update metrics based on a batch of audio.
 
     Args:
@@ -228,7 +228,7 @@ def __init__(self, sample_rate, frame_rate, name='f0_crepe'):
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, audio_gen):
+  def update_state(self, batch, audio_gen):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
     """Update metrics based on a batch of audio.
 
     Args:
@@ -293,7 +293,7 @@ def __init__(
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, f0_hz_predict):
+  def update_state(self, batch, f0_hz_predict):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
     """Update metrics based on a batch of audio.
 
     Args:
