Update imports

Author: kesmit13

singlestoredb/functions/typing/numpy.py

@@ -6,26 +6,6 @@
 import numpy.typing as npt
 from numpy import array  # noqa: F401
 
-NDArray = npt.NDArray
-
-StringArray = StrArray = npt.NDArray[np.str_]
-BytesArray = npt.NDArray[np.bytes_]
-Float16Array = HalfArray = npt.NDArray[np.float16]
-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]
-JSONArray = npt.NDArray[np.object_]
-=======
 try:
     from typing import TypeAlias  # type: ignore
 except ImportError:
@@ -48,6 +28,11 @@
     npt.NDArray[np.bytes_], UDFAttrs(sql_type=sql_types.BLOB(nullable=False)),
 ]
 
+Float16Array: TypeAlias = Annotated[
+    npt.NDArray[np.float16], UDFAttrs(sql_type=sql_types.FLOAT16(nullable=False)),
+]
+HalfArray: TypeAlias = Float16Array
+
 Float32Array: TypeAlias = Annotated[
     npt.NDArray[np.float32], UDFAttrs(sql_type=sql_types.FLOAT(nullable=False)),
 ]
@@ -128,4 +113,3 @@ def default(self, obj: Any) -> Any:
 
 
 __all__ = ['array'] + [x for x in globals().keys() if x.endswith('Array')]
->>>>>>> fd486846 (Implement transformers for UDF inputs and outputs)

singlestoredb/tests/ext_funcs/__init__.py

@@ -16,15 +16,15 @@
 from singlestoredb.functions import Masked
 from singlestoredb.functions import Table
 from singlestoredb.functions import udf
-from singlestoredb.functions.dtypes import BIGINT
-from singlestoredb.functions.dtypes import BLOB
-from singlestoredb.functions.dtypes import BOOL
-from singlestoredb.functions.dtypes import DOUBLE
-from singlestoredb.functions.dtypes import FLOAT
-from singlestoredb.functions.dtypes import MEDIUMINT
-from singlestoredb.functions.dtypes import SMALLINT
-from singlestoredb.functions.dtypes import TEXT
-from singlestoredb.functions.dtypes import TINYINT
+from singlestoredb.functions.sql_types import BIGINT
+from singlestoredb.functions.sql_types import BLOB
+from singlestoredb.functions.sql_types import BOOL
+from singlestoredb.functions.sql_types import DOUBLE
+from singlestoredb.functions.sql_types import FLOAT
+from singlestoredb.functions.sql_types import MEDIUMINT
+from singlestoredb.functions.sql_types import SMALLINT
+from singlestoredb.functions.sql_types import TEXT
+from singlestoredb.functions.sql_types import TINYINT
 from singlestoredb.functions.typing import JSON
 from singlestoredb.functions.typing import numpy as npt
 from singlestoredb.functions.typing import pandas as pdt