mypy and docs

Lightning-AI · Nov 12, 2022 · 3ab20e8 · 3ab20e8
1 parent 1de7ff6
commit 3ab20e8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/pytorch_lightning/loggers/csv_logs.py b/src/pytorch_lightning/loggers/csv_logs.py
@@ -133,7 +133,7 @@ def __init__(
         flush_logs_every_n_steps: int = 100,
     ):
         super().__init__()
-        self._save_dir = save_dir
+        self._save_dir = os.fspath(save_dir)
         self._name = name or ""
         self._version = version
         self._prefix = prefix

diff --git a/src/pytorch_lightning/loggers/tensorboard.py b/src/pytorch_lightning/loggers/tensorboard.py
@@ -43,6 +43,8 @@
 _TENSORBOARD_AVAILABLE = RequirementCache("tensorboard>=2.9.1")
 if TYPE_CHECKING and _TENSORBOARD_AVAILABLE:
     from torch.utils.tensorboard import SummaryWriter
+else:
+    SummaryWriter = Any
 
 
 class TensorBoardLogger(Logger):