diff --git a/python/gen_requirements.py b/python/gen_requirements.py
index 9778937ae80b..7280a787ffea 100755
--- a/python/gen_requirements.py
+++ b/python/gen_requirements.py
@@ -72,6 +72,7 @@
                 "scipy",
                 "synr",
                 "tornado",
+                "typing_extensions",
             ],
         ),
     ),
@@ -277,6 +278,7 @@
     ("torch", None),
     ("torchvision", None),
     ("tornado", None),
+    ("typing_extensions", None),
     ("xgboost", ">=1.1.0,<1.6.0"),  # From PR #4953 & Issue #12009
 ]