Add tensorboard.

aistairc · Jan 31, 2025 · fbfb63e · fbfb63e
1 parent dc344ca
commit fbfb63e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",