diff --git a/docs/source/conf.py b/docs/source/conf.py
index 9716cf1..4e9d888 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -61,4 +61,5 @@
     "bioneuralnet.external_tools",
     "sklearn",
     "statsmodels",
+    "xgboost",
 ]
diff --git a/requirements.txt b/requirements.txt
index 3fae74f..e43b72d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,5 +7,6 @@ networkx
 python-louvain
 ray[tune]
 statsmodels
+xgboost
 # torch
 # torch_geometric
\ No newline at end of file
diff --git a/setup.cfg b/setup.cfg
index 296ed31..f96e703 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -29,6 +29,7 @@ install_requires =
     statsmodels
     networkx
     python-louvain
+    xgboost
     ray[tune]
 
 [options.packages.find]