add types

Author: attack68

pandas/io/formats/style.py

@@ -18,12 +18,12 @@
     Optional,
     Sequence,
     Tuple,
+    TypedDict,
     Union,
 )
 from uuid import uuid4
 
 import numpy as np
-from typing_extensions import TypedDict
 
 from pandas._config import get_option
 
@@ -1233,19 +1233,19 @@ def set_table_styles(
                 obj, idf = self.data.index, ".row"
 
             table_styles = [
-                {
-                    "selector": str(s["selector"]) + idf + str(obj.get_loc(key)),
-                    "props": _maybe_convert_css_to_tuples(s["props"]),
-                }
+                CSSDict(
+                    selector=str(s["selector"]) + idf + str(obj.get_loc(key)),
+                    props=_maybe_convert_css_to_tuples(s["props"]),
+                )
                 for key, styles in table_styles.items()
                 for s in styles
             ]
         else:
             table_styles = [
-                {
-                    "selector": s["selector"],
-                    "props": _maybe_convert_css_to_tuples(s["props"]),
-                }
+                CSSDict(
+                    selector=s["selector"],
+                    props=_maybe_convert_css_to_tuples(s["props"]),
+                )
                 for s in table_styles
             ]
 
@@ -1875,10 +1875,10 @@ def _class_styles(self):
         styles : List
         """
         return [
-            {
-                "selector": f".{self.class_name}",
-                "props": _maybe_convert_css_to_tuples(self.class_properties),
-            }
+            CSSDict(
+                selector=f".{self.class_name}",
+                props=_maybe_convert_css_to_tuples(self.class_properties),
+            )
         ]
 
     def _pseudo_css(self, uuid: str, name: str, row: int, col: int, text: str):
@@ -1912,27 +1912,16 @@ def _pseudo_css(self, uuid: str, name: str, row: int, col: int, text: str):
         -------
         pseudo_css : List
         """
+        selector_id = "#T_" + uuid + "row" + str(row) + "_col" + str(col)
         return [
-            {
-                "selector": "#T_"
-                + uuid
-                + "row"
-                + str(row)
-                + "_col"
-                + str(col)
-                + f":hover .{name}",
-                "props": [("visibility", "visible")],
-            },
-            {
-                "selector": "#T_"
-                + uuid
-                + "row"
-                + str(row)
-                + "_col"
-                + str(col)
-                + f" .{name}::after",
-                "props": [("content", f'"{text}"')],
-            },
+            CSSDict(
+                selector=selector_id + f":hover .{name}",
+                props=[("visibility", "visible")],
+            ),
+            CSSDict(
+                selector=selector_id + f" .{name}::after",
+                props=[("content", f'"{text}"')],
+            ),
         ]
 
     def _translate(self, styler_data: FrameOrSeriesUnion, uuid: str, d: Dict):