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))