Silence some pytype errors.

PiperOrigin-RevId: 528510090
magenta · May 1, 2023 · b2d180a · b2d180a
1 parent 0188381
commit b2d180a
Showing 1 changed file with 3 additions and 3 deletions.
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: