From 2e48b3e8725326abd3e9cf82718f7d6debdd8297 Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Thu, 11 Jul 2024 16:41:50 +0530 Subject: [PATCH] fix: Fixed the `1st argument` name in classmethods (#31907) Fixed the first argument name in few classmethods. --- .../models/layoutlmv2/configuration_layoutlmv2.py | 2 +- tests/optimization/test_optimization.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py b/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py index db1fdf7da2..d2a9d37bd1 100644 --- a/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py +++ b/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py @@ -177,7 +177,7 @@ class LayoutLMv2Config(PretrainedConfig): ) @classmethod - def get_default_detectron2_config(self): + def get_default_detectron2_config(cls): return { "MODEL.MASK_ON": True, "MODEL.PIXEL_STD": [57.375, 57.120, 58.395], diff --git a/tests/optimization/test_optimization.py b/tests/optimization/test_optimization.py index 5240b03779..adf3039bb4 100644 --- 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))