Fixed mypy issues (#3129)

* Fixed mypy issues * Revert changes for visdom
pytorch · Nov 10, 2023 · b7c9620 · b7c9620
1 parent 39e46bf
commit b7c9620
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ignite/contrib/engines/common.py b/ignite/contrib/engines/common.py
@@ -167,9 +167,7 @@ def _setup_common_training_handlers(
 
     if lr_scheduler is not None:
         if isinstance(lr_scheduler, PyTorchLRScheduler):
-            trainer.add_event_handler(
-                Events.ITERATION_COMPLETED, lambda engine: cast(PyTorchLRScheduler, lr_scheduler).step()
-            )
+            trainer.add_event_handler(Events.ITERATION_COMPLETED, lambda engine: lr_scheduler.step())
         else:
             trainer.add_event_handler(Events.ITERATION_STARTED, lr_scheduler)