diff --git a/ignite/contrib/handlers/base_logger.py b/ignite/contrib/handlers/base_logger.py index e8825b45005b..bb4a1181382f 100644 --- a/ignite/contrib/handlers/base_logger.py +++ b/ignite/contrib/handlers/base_logger.py @@ -167,7 +167,7 @@ def attach(self, engine, log_handler, event_name): return engine.add_event_handler(event_name, log_handler, self, name) - def attach_output_handler(self, engine: Engine, event_name: str, *args: Any, **kwargs: Mapping): + def attach_output_handler(self, engine: Engine, event_name: Any, *args: Any, **kwargs: Mapping): """Shortcut method to attach `OutputHandler` to the logger. Args: @@ -182,7 +182,7 @@ def attach_output_handler(self, engine: Engine, event_name: str, *args: Any, **k """ return self.attach(engine, self._create_output_handler(*args, **kwargs), event_name=event_name) - def attach_opt_params_handler(self, engine: Engine, event_name: str, *args: Any, **kwargs: Mapping): + def attach_opt_params_handler(self, engine: Engine, event_name: Any, *args: Any, **kwargs: Mapping): """Shortcut method to attach `OptimizerParamsHandler` to the logger. Args: