Verbose fix (#2957)

* remove codecov

* RankProcessFirst

* annotations

* from class to contextlib

* from class to contextlib and test

* del test file

* uniq folder for test

* refactor tests + new assert_test

* add to __all__, remove idist import

* Apply suggestions from code review

* Apply suggestions from code review

* Update tests/ignite/distributed/utils/test_native.py

* Added local arg and renamed function

* verbose del

* remove cast

---------

Co-authored-by: vfdev <[email protected]>

Author: AlexanderChaptykov

ignite/handlers/param_scheduler.py

@@ -809,12 +809,8 @@ def optimizer(self) -> torch.optim.Optimizer:
         return self._lr_scheduler.optimizer
 
     def get_lr(self, epoch: Optional[int] = None) -> List[float]:
-        # TODO: Remove this workaround when pytorch has fixed wrong type hints:
-        # https://github.com/pytorch/pytorch/pull/102067
-        # Replace below T_mult -> self._lr_scheduler.T_mult
-        # Replace below eta_min -> self._lr_scheduler.eta_min
-        T_mult = cast(int, self._lr_scheduler.T_mult)
-        eta_min = cast(float, self._lr_scheduler.eta_min)
+        T_mult = self._lr_scheduler.T_mult
+        eta_min = self._lr_scheduler.eta_min
 
         if epoch is None and self.last_epoch < 0:
             epoch = 0

tests/ignite/handlers/test_param_scheduler.py

@@ -1378,7 +1378,7 @@ def get_optim():
         return torch.optim.SGD([t1], lr=lr)
 
     def get_cos_shed():
-        return CosineAnnealingWarmRestarts(optimizer, T_0=T_0, T_mult=T_mult, verbose=False)
+        return CosineAnnealingWarmRestarts(optimizer, T_0=T_0, T_mult=T_mult)
 
     optimizer = get_optim()
     scheduler = get_cos_shed()