Use generic in xgboost model doc decorator

python-package/xgboost/sklearn.py

@@ -536,13 +536,14 @@ 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 +569,7 @@ def get_doc(item: str) -> str:
         }
         return __doc[item]
 
-    def adddoc(cls: Type) -> Type:
+    def adddoc(cls: TDoc) -> TDoc:
         doc = [
             """
 Parameters