Fix typing issues

Author: kesmit13

singlestoredb/functions/ext/rowdat_1.py

@@ -578,7 +578,10 @@ def _load_numpy_accel(
             # then fill it in with the transformed values. The transformer may have
             # an output_type attribute that we can use to create a more specific type.
             if getattr(transformer, 'output_type', None):
-                new_col = np.empty(len(numpy_cols[i][0]), dtype=transformer.output_type)
+                new_col = np.empty(
+                    len(numpy_cols[i][0]),
+                    dtype=transformer.output_type,  # type: ignore
+                )
                 new_col[:] = list(map(transformer, numpy_cols[i][0]))
             else:
                 new_col = np.array(list(map(transformer, numpy_cols[i][0])))

singlestoredb/functions/typing/__init__.py

@@ -53,14 +53,17 @@ def __new__(cls, *args: Unpack[Ts]) -> 'Table[Tuple[Unpack[Ts]]]':  # type: igno
         return tuple.__new__(cls, args)  # type: ignore
 
 
-class Transformer(Protocol):
+class TypedTransformer(Protocol):
 
     output_type: Optional[Any] = None
 
     def __call__(self, *args: Any, **kwargs: Any) -> Any:
         ...
 
 
+Transformer = Union[Callable[..., Any], TypedTransformer]
+
+
 def output_type(dtype: Any) -> Callable[..., Any]:
     """
     Decorator that sets the output_type attribute on a function.

singlestoredb/functions/typing/numpy.py

@@ -7,9 +7,9 @@
 from numpy import array  # noqa: F401
 
 try:
-    from typing import TypeAlias
+    from typing import TypeAlias  # type: ignore
 except ImportError:
-    from typing_extensions import TypeAlias
+    from typing_extensions import TypeAlias  # type: ignore
 
 from . import UDFAttrs
 from . import json_or_null_dumps

singlestoredb/functions/typing/pandas.py

@@ -8,9 +8,9 @@
 from pandas import Series  # noqa: F401
 
 try:
-    from typing import TypeAlias
+    from typing import TypeAlias  # type: ignore
 except ImportError:
-    from typing_extensions import TypeAlias
+    from typing_extensions import TypeAlias  # type: ignore
 
 from . import UDFAttrs
 from . import json_or_null_dumps

singlestoredb/functions/typing/polars.py

@@ -7,9 +7,9 @@
 from polars import Series  # noqa: F401
 
 try:
-    from typing import TypeAlias
+    from typing import TypeAlias  # type: ignore
 except ImportError:
-    from typing_extensions import TypeAlias
+    from typing_extensions import TypeAlias  # type: ignore
 
 from . import UDFAttrs
 from . import json_or_null_dumps

singlestoredb/functions/typing/pyarrow.py

@@ -8,9 +8,9 @@
 from pyarrow import Table  # noqa: F401
 
 try:
-    from typing import TypeAlias
+    from typing import TypeAlias  # type: ignore
 except ImportError:
-    from typing_extensions import TypeAlias
+    from typing_extensions import TypeAlias  # type: ignore
 
 from . import UDFAttrs
 from . import json_or_null_dumps