From 5e11924ca6aaea7898caca94675f41f67706a406 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm 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(-) 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