Automated Code Change: Suppress pytype errors

PiperOrigin-RevId: 700257299
tensorflow · Nov 26, 2024 · ccfe1e4 · ccfe1e4
1 parent 8c4ab9b
commit ccfe1e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lingvo/base_runner.py b/lingvo/base_runner.py
@@ -176,7 +176,7 @@ def _ExportMetrics(self, **kwargs):
     pass
 
   def _ShouldEarlyStop(self, sess: Optional[tf.Session] = None) -> bool:
-    return self._early_stop and self._early_stop.Stop(sess)
+    return self._early_stop and self._early_stop.Stop(sess)  # pytype: disable=bad-return-type
 
   def _ShouldStop(self, sess=None, step=None, check_early_stop=True) -> bool:
     """Check if the runner should stop.