ENH: Improve Styler

Author: bashtage

pandas-stubs/io/formats/style.pyi

@@ -3,6 +3,7 @@ from typing import (
     Callable,
     Hashable,
     Sequence,
+    TypeVar,
 )
 
 from pandas.core.frame import DataFrame
@@ -25,6 +26,8 @@ from pandas.io.formats.style_render import (
     Subset,
 )
 
+_StylerT = TypeVar("_StylerT", bound=Styler)
+
 class Styler(StylerRenderer):
     def __init__(
         self,
@@ -252,8 +255,13 @@ class Styler(StylerRenderer):
     @classmethod
     def from_custom_template(
         cls,
-        searchpath,
+        searchpath: str | list[str],
         html_table: str | None = ...,
         html_style: str | None = ...,
-    ): ...
-    def pipe(self, func: Callable, *args, **kwargs): ...
+    ) -> _StylerT: ...
+    def pipe(
+        self,
+        func: Callable[[Styler], Styler] | tuple[Callable[[Styler], Styler], str],
+        *args: Any,
+        **kwargs: Any,
+    ) -> Styler: ...