From 3fe87df645a6a793bf3e7c5690ff8d0e357b74e3 Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Wed, 13 May 2020 17:08:55 +0200 Subject: [PATCH] Fixed typing in attach_opt_params_handler and attach_output_handler --- ignite/contrib/handlers/base_logger.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: