From 5e11924ca6aaea7898caca94675f41f67706a406 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 5 Apr 2021 23:25:25 +0200
Subject: Fix typing

---
 text_recognizer/models/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'text_recognizer/models/base.py')

diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py
index 11d1eb1..f4df3f3 100644
--- a/text_recognizer/models/base.py
+++ b/text_recognizer/models/base.py
@@ -40,7 +40,7 @@ class LitBaseModel(pl.LightningModule):
         args = {} or criterion.args
         return getattr(nn, criterion.type)(**args)
 
-    def _configure_optimizer(self) -> torch.optim.Optimizer:
+    def _configure_optimizer(self) -> Type[torch.optim.Optimizer]:
         """Configures the optimizer."""
         args = {} or self._optimizer.args
         if self._optimizer.type == "MADGRAD":
-- 
cgit v1.2.3-70-g09d2