diff options
| -rw-r--r-- | noxfile.py | 1 | ||||
| -rw-r--r-- | training/run_experiment.py | 2 |
2 files changed, 1 insertions, 2 deletions
@@ -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() |