Add tensorboard import message

This commit is contained in:
mbsantiago 2026-08-08 13:50:36 +01:00
parent f4a0421ef3
commit 981b625c34

View File

@ -201,7 +201,14 @@ def create_tensorboard_logger(
experiment_name: str | None = None,
run_name: str | None = None,
) -> 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:
log_dir = Path(config.log_dir)