Created mappings

1 files changed, 4 insertions, 2 deletions

diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py
index c6d5d73..aeda039 100644
--- a/text_recognizer/models/base.py
+++ b/text_recognizer/models/base.py
@@ -49,7 +49,9 @@ class LitBaseModel(pl.LightningModule):
             optimizer_class = getattr(torch.optim, self._optimizer.type)
         return optimizer_class(params=self.parameters(), **args)
 
-    def _configure_lr_scheduler(self, optimizer: Type[torch.optim.Optimizer]) -> Dict[str, Any]:
+    def _configure_lr_scheduler(
+        self, optimizer: Type[torch.optim.Optimizer]
+    ) -> Dict[str, Any]:
         """Configures the lr scheduler."""
         scheduler = {"monitor": self.monitor}
         args = {} or self._lr_scheduler.args
@@ -59,7 +61,7 @@ class LitBaseModel(pl.LightningModule):
 
         scheduler["scheduler"] = getattr(
             torch.optim.lr_scheduler, self._lr_scheduler.type
-            )(optimizer, **args)
+        )(optimizer, **args)
 
         return scheduler