ENH: Improve Styler typing (#282)

* ENH: Improve styler typing

* TYP: Add typing for StylerRenderer

* ENH: Improve Styler

* ENH: Improve definition os subset

* BUG: Overcome 3.8 limit

* BUG: Fix IndexSlice type

* MAINT: Set backend for MPL

* TYP: Simplify types

* TYP: Improve axis

* TYP: Remove attributes

* TYP: Final small fixes

* TYP: Small fixes for Styler pipe

Co-authored-by: Kevin Sheppard <kevin.sheppard@gmail.com>

Author: bashtage

.github/workflows/test.yml

@@ -2,6 +2,10 @@ name: 'Test'
 
 on:
   [push, pull_request, workflow_dispatch]
+
+env:
+  MPLBACKEND: 'Agg'
+
 jobs:
   released:
     runs-on: ${{ matrix.os }}

pandas-stubs/_typing.pyi

@@ -20,6 +20,7 @@ from typing import (
     Optional,
     Protocol,
     Sequence,
+    TypedDict,
     TypeVar,
     Union,
 )
@@ -249,4 +250,14 @@ FileWriteMode = Literal[
 ]
 ColspaceArgType = str | int | Sequence[int | str] | Mapping[Hashable, str | int]
 
+class StyleExportDict(TypedDict, total=False):
+    apply: Any
+    table_attributes: Any
+    table_styles: Any
+    hide_index: bool
+    hide_columns: bool
+    hide_index_names: bool
+    hide_column_names: bool
+    css: dict[str, str | int]
+
 __all__ = ["npt", "type_t"]

pandas-stubs/core/indexing.pyi

@@ -1,3 +1,9 @@
+from typing import (
+    Generic,
+    TypeVar,
+    Union,
+)
+
 import numpy as np
 from pandas.core.indexes.api import Index
 
@@ -7,8 +13,10 @@ from pandas._typing import (
     StrLike,
 )
 
-class _IndexSlice:
-    def __getitem__(self, arg) -> tuple[StrLike | Scalar | slice, ...]: ...
+_IndexSliceT = TypeVar("_IndexSliceT", bound=Union[StrLike, Scalar, slice])
+
+class _IndexSlice(Generic[_IndexSliceT]):
+    def __getitem__(self, arg) -> tuple[_IndexSliceT, ...]: ...
 
 IndexSlice: _IndexSlice