From fbfb63e11439dca785aa20167c235a342b635add Mon Sep 17 00:00:00 2001
From: KanaiYuma <kanai-itc1013@aist.go.jp>
Date: Fri, 31 Jan 2025 10:40:44 +0900
Subject: [PATCH] Add tensorboard.

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 2e178b55..b0370877 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
     "numpy",
     "optuna>=3.4.0",
 ]
-torch_dependencies = ["hydra-core", "lightning>=2.2.1", "torch>=2.2.0", "h5py"]
+torch_dependencies = ["hydra-core", "lightning>=2.2.1", "torch>=2.2.0", "h5py", "tensorboard"]
 dev_dependencies = [
     "mypy",
     "myst-parser",