feat(typing): Adds RowCol generic

Related vega#3536 (comment)

altair/vegalite/v5/schema/_typing.py

@@ -170,6 +170,28 @@ def is_color_hex(obj: Any) -> TypeIs[ColorHex]:
     return bool(pattern.fullmatch(obj))
 
 
+CT = TypeVar("CT")
+RT = TypeVar("RT")
+
+
+class RowCol(TypedDict, Generic[CT, RT], total=False):
+    """
+    A `Generic`_ two-item ``dict``.
+
+    Parameters
+    ----------
+    column: CT
+    row: RT
+
+    .. _Generic:
+        https://typing.readthedocs.io/en/latest/spec/generics.html#generics
+
+    """
+
+    column: CT
+    row: RT
+
+
 VegaThemes: TypeAlias = Literal[
     "carbong10",
     "carbong100",

tools/generate_schema_wrapper.py

@@ -352,6 +352,26 @@ def is_color_hex(obj: Any) -> TypeIs[ColorHex]:
     pattern: re.Pattern[str] = next(it)
     return bool(pattern.fullmatch(obj))
 
+
+CT = TypeVar("CT")
+RT = TypeVar("RT")
+
+class RowCol(TypedDict, Generic[CT, RT], total=False):
+    """
+    A `Generic`_ two-item ``dict``.
+
+    Parameters
+    ----------
+    column: CT
+    row: RT
+
+    .. _Generic:
+        https://typing.readthedocs.io/en/latest/spec/generics.html#generics
+
+    """
+
+    column: CT
+    row: RT
 '''
 
 _ChannelType = Literal["field", "datum", "value"]

tools/schemapi/utils.py

@@ -556,6 +556,10 @@ def title_to_type_reprs(self, *, use_concrete: bool) -> set[str]:
             value = self.properties["value"]
             t = value.to_type_repr(target="annotation", use_concrete=use_concrete)
             tps.add(f"Value[{t}]")
+        elif self.is_rowcol():
+            row = self.properties["row"]
+            t = row.to_type_repr(target="annotation", use_concrete=use_concrete)
+            tps.add(f"RowCol[{t}, {t}]")
         elif title in REMAP_TITLE:
             tps.update(REMAP_TITLE[title])
         elif (
@@ -714,6 +718,14 @@ def is_object(self) -> bool:
     def is_value(self) -> bool:
         return self.is_object() and self.properties.keys() == {"value"}
 
+    def is_rowcol(self) -> bool:
+        props = self.properties
+        return (
+            self.is_object()
+            and props.keys() == {"column", "row"}
+            and props["column"] == props["row"]
+        )
+
     def is_array(self) -> bool:
         return self.type == "array"