Merge pull request #15 from FlorentFlament/main

Fix #14, "requires tensorboard to be installed" RuntimeError
ibm-granite · Mar 3, 2024 · 98bfa05 · 98bfa05
2 parents d07c741 + 837023c
commit 98bfa05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ packages = ["tsfm_public", "tsfm_public.toolkit", "tsfmhfdemos"]
 
 [project.optional-dependencies]
 
-notebooks = ["jupyter", "matplotlib", "datasets", "ipywidgets", "plotly", "kaleido"]
+notebooks = ["jupyter", "matplotlib", "datasets", "ipywidgets", "plotly", "kaleido", "tensorboard"]
 testing = ["pytest", "tsfm_public[notebooks]"]
 dev = ["pre-commit", "tsfm_public[testing]", "ruff==0.1.5"]
 evaluation = [