From 981b625c34224eaa14e1a7840bb0e586b7e6c9cc Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Sat, 8 Aug 2026 13:50:36 +0100
Subject: [PATCH] Add tensorboard import message

---
 src/batdetect2/logging.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/batdetect2/logging.py b/src/batdetect2/logging.py
index b128451..2d5c728 100644
--- a/src/batdetect2/logging.py
+++ b/src/batdetect2/logging.py
@@ -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)