diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py
index b197539bfc1f..62d91be29bac 100644
--- a/python-package/xgboost/sklearn.py
+++ b/python-package/xgboost/sklearn.py
@@ -536,13 +536,15 @@ def task(i: int) -> float:
             information) instead.
 """
 
+TDoc = TypeVar("TDoc", bound=Type)
+
 
 def xgboost_model_doc(
     header: str,
     items: List[str],
     extra_parameters: Optional[str] = None,
     end_note: Optional[str] = None,
-) -> Callable[[Type], Type]:
+) -> Callable[[TDoc], TDoc]:
     """Obtain documentation for Scikit-Learn wrappers
 
     Parameters
@@ -568,7 +570,7 @@ def get_doc(item: str) -> str:
         }
         return __doc[item]
 
-    def adddoc(cls: Type) -> Type:
+    def adddoc(cls: TDoc) -> TDoc:
         doc = [
             """
 Parameters