[tune] added type hints (#10806)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
2026-08-18 12:20:14 +08:00 · 2020-09-15 21:03:56 -07:00
parent 5e030db8a5
commit c9fafe7733
31 changed files with 709 additions and 472 deletions
@@ -39,7 +39,7 @@ class TuneCallback(Callback):
                    on, self._allowed))
        self._on = on

-    def _handle(self, logs: Dict):
+    def _handle(self, logs: Dict, when: str):
        raise NotImplementedError

    def on_batch_begin(self, batch, logs=None):