diff --git a/setup.py b/setup.py
index c1b961d..f6c1223 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,7 @@ setup(
         'pydantic',
         'matplotlib',
         'python-rapidjson',
+        'tensorboard',
     ],
 
     test_suite='tests',