diff --git a/pytorch_optimizer/lr_scheduler/rex.py b/pytorch_optimizer/lr_scheduler/rex.py index 5d7ad3d31..2c6974911 100644 --- a/pytorch_optimizer/lr_scheduler/rex.py +++ b/pytorch_optimizer/lr_scheduler/rex.py @@ -52,7 +52,7 @@ def get_linear_lr(self) -> float: return self.min_lr + (self.max_lr - self.min_lr) * ((1.0 - progress) / (1.0 - progress / 2.0)) - def step(self, epoch: Optional[int] = None) -> float: + def step(self, epoch: Optional[int] = None) -> float: # noqa: ARG002 value: float = self.get_linear_lr() self.step_t += 1