diff --git a/setup.py b/setup.py
index ff59e7a66..9c717df8b 100644
--- a/setup.py
+++ b/setup.py
@@ -12,9 +12,11 @@ hyperopt = [
 
 freqai = [
     'scikit-learn',
+    'joblib',
     'catboost; platform_machine != "aarch64"',
     'lightgbm',
-    'xgboost'
+    'xgboost',
+    'tensorboard'
 ]
 
 freqai_rl = [