diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt
index 6b5dd7a5..d0021f64 100644
--- a/docs/requirements-docs.txt
+++ b/docs/requirements-docs.txt
@@ -3,7 +3,7 @@ pandas>=1.5
 matplotlib>=3.6
 nbval
 scikit-learn>=0.22.1
-xgboost>=1
+xgboost<2.1
 lightgbm
 sphinx==5.3.0
 nbsphinx
diff --git a/requirements-dev.txt b/requirements-dev.txt
index 9a1308f0..d869748c 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -11,7 +11,7 @@ tqdm<5
 # Extras
 #
 scikit-learn>=1.3,<1.4
-xgboost>=0.90,<2
+xgboost>=0.90,<2.1
 lightgbm>=2,<4
 
 # PyTorch doesn't support Python 3.11 yet (pytorch/pytorch#86566)
diff --git a/setup.py b/setup.py
index 4a3334e0..ce57e9f9 100644
--- a/setup.py
+++ b/setup.py
@@ -54,7 +54,7 @@
     long_description = "\n".join(lines[last_html_index:])
 
 extras = {
-    "xgboost": ["xgboost>=0.90,<2"],
+    "xgboost": ["xgboost>=0.90, <2.1"],
     "scikit-learn": ["scikit-learn>=1.3,<1.4"],
     "lightgbm": ["lightgbm>=2,<4"],
     "pytorch": [