fix: Fixed the 1st argument name in classmethods (#31907)

Fixed the first argument name in few classmethods.
2024-07-11 16:41:50 +05:30
parent 48c20700e1
commit 2e48b3e872
2 changed files with 3 additions and 3 deletions
--- a/tests/optimization/test_optimization.py
+++ b/tests/optimization/test_optimization.py
@@ -209,5 +209,5 @@ class LambdaScheduleWrapper:
        return self.fn(*args, **kwargs)

    @classmethod
-    def wrap_scheduler(self, scheduler):
-        scheduler.lr_lambdas = list(map(self, scheduler.lr_lambdas))
+    def wrap_scheduler(cls, scheduler):
+        scheduler.lr_lambdas = list(map(cls, scheduler.lr_lambdas))