Add type aliases

kesmit13 · kesmit13 · commit 72ee27bb5672 · 2025-06-06T10:50:35.000-05:00
diff --git a/singlestoredb/functions/typing/__init__.py b/singlestoredb/functions/typing/__init__.py
diff --git a/singlestoredb/functions/typing/numpy.py b/singlestoredb/functions/typing/numpy.py
@@ -0,0 +1,20 @@
+import numpy as np
+import numpy.typing as npt
+
+NDArray = npt.NDArray
+
+StringArray = StrArray = npt.NDArray[np.str_]
+BytesArray = npt.NDArray[np.bytes_]
+Float32Array = FloatArray = npt.NDArray[np.float32]
+Float64Array = DoubleArray = npt.NDArray[np.float64]
+IntArray = npt.NDArray[np.int_]
+Int8Array = npt.NDArray[np.int8]
+Int16Array = npt.NDArray[np.int16]
+Int32Array = npt.NDArray[np.int32]
+Int64Array = npt.NDArray[np.int64]
+UInt8Array = npt.NDArray[np.uint8]
+UInt16Array = npt.NDArray[np.uint16]
+UInt32Array = npt.NDArray[np.uint32]
+UInt64Array = npt.NDArray[np.uint64]
+DateTimeArray = npt.NDArray[np.datetime64]
+TimeDeltaArray = npt.NDArray[np.timedelta64]
diff --git a/singlestoredb/functions/typing/pandas.py b/singlestoredb/functions/typing/pandas.py
@@ -0,0 +1,2 @@
+from pandas import DataFrame  # noqa: F401
+from pandas import Series  # noqa: F401
diff --git a/singlestoredb/functions/typing/polars.py b/singlestoredb/functions/typing/polars.py
@@ -0,0 +1,2 @@
+from polars import DataFrame  # noqa: F401
+from polars import Series  # noqa: F401
diff --git a/singlestoredb/functions/typing/pyarrow.py b/singlestoredb/functions/typing/pyarrow.py
@@ -0,0 +1,2 @@
+from pyarrow import Array  # noqa: F401
+from pyarrow import Table  # noqa: F401
diff --git a/singlestoredb/tests/ext_funcs/__init__.py b/singlestoredb/tests/ext_funcs/__init__.py

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+from pandas import DataFrame # noqa: F401`
	`2`	`+from pandas import Series # noqa: F401`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+from polars import DataFrame # noqa: F401`
	`2`	`+from polars import Series # noqa: F401`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+from pyarrow import Array # noqa: F401`
	`2`	`+from pyarrow import Table # noqa: F401`