lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-08-22 03:00:10 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add tensorboard import message

Browse Source
This commit is contained in:
mbsantiago 2026-08-08 13:50:36 +01:00
parent f4a0421ef3
commit 981b625c34
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/batdetect2/logging.py
View File

@ -201,7 +201,14 @@ def create_tensorboard_logger(
experiment_name: str | None = None, experiment_name: str | None = None,
run_name: str | None = None, run_name: str | None = None,
) -> Logger: ) -> Logger:
from lightning.pytorch.loggers import TensorBoardLogger try:
from lightning.pytorch.loggers import TensorBoardLogger
except ImportError as error:
raise ValueError(
"TensorBoard is not installed and cannot be used for logging. "
"Make sure you have it installed by running `pip install tensorboard` "
"or `uv add tensorboard`"
) from error
if log_dir is None: if log_dir is None:
log_dir = Path(config.log_dir) log_dir = Path(config.log_dir)