style: fix ARG002

kozistr · Jun 29, 2024 · 60e3848 · 60e3848
1 parent bba627d
commit 60e3848
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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