diff --git a/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py b/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py index db1fdf7da2aa2c..d2a9d37bd12a87 100644 --- a/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py +++ b/src/transformers/models/layoutlmv2/configuration_layoutlmv2.py @@ -177,7 +177,7 @@ def __init__( ) @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 5240b03779b735..adf3039bb45033 100644 --- a/tests/optimization/test_optimization.py +++ b/tests/optimization/test_optimization.py @@ -209,5 +209,5 @@ def __call__(self, *args, **kwargs): 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))