summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-04-07 22:26:29 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-04-07 22:26:29 +0200
commitf4629a0d4149d5870c9fd8ce83ff5d391bd7ddd3 (patch)
tree3d4364a53a05a9b019b3af430d3092d434e2be1c /training
parent8afa8e1c6e9623b0dea86236da04b2b4173e9443 (diff)
Typing fixed, tasks removed from noxfile
Diffstat (limited to 'training')
-rw-r--r--training/run_experiment.py2
1 files changed, 1 insertions, 1 deletions
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()