From f4629a0d4149d5870c9fd8ce83ff5d391bd7ddd3 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 7 Apr 2021 22:26:29 +0200
Subject: Typing fixed, tasks removed from noxfile

---
 noxfile.py                 | 1 -
 training/run_experiment.py | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/noxfile.py b/noxfile.py
index 0e2ac7b..1976a08 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -11,7 +11,6 @@ nox.options.sessions = "lint", "mypy", "pytype", "safety", "tests"
 locations = (
     "text_recognizer",
     "training",
-    "tasks",
     "tests",
     "noxfile.py",
     "docs/conf.py",
diff --git a/training/run_experiment.py b/training/run_experiment.py
index 1fb4bc2..491112c 100644
--- a/training/run_experiment.py
+++ b/training/run_experiment.py
@@ -69,7 +69,7 @@ def _configure_callbacks(
 
 def _configure_logger(
     network: Type[nn.Module], args: Dict, use_wandb: bool
-) -> pl.loggers.WandbLogger:
+) -> Type[pl.loggers.LightningLoggerBase]:
     """Configures lightning logger."""
     if use_wandb:
         pl_logger = pl.loggers.WandbLogger()
-- 
cgit v1.2.3-70-g09d2