fix: reexport new types (zen-xu#199)

Author: zen-xu

pyarrow-stubs/__init__.pyi

@@ -56,6 +56,8 @@ from pyarrow.lib import (
     large_binary,
     large_string,
     large_utf8,
+    decimal32,
+    decimal64,
     decimal128,
     decimal256,
     list_,
@@ -69,7 +71,11 @@ from pyarrow.lib import (
     dense_union,
     dictionary,
     run_end_encoded,
+    json_,
+    uuid,
     fixed_shape_tensor,
+    bool8,
+    opaque,
     field,
     type_for_alias,
     DataType,
@@ -97,6 +103,10 @@ from pyarrow.lib import (
     ExtensionType,
     RunEndEncodedType,
     FixedShapeTensorType,
+    Bool8Type,
+    UuidType,
+    JsonType,
+    OpaqueType,
     PyExtensionType,
     UnknownExtensionType,
     register_extension_type,
@@ -164,6 +174,10 @@ from pyarrow.lib import (
     ExtensionArray,
     RunEndEncodedArray,
     FixedShapeTensorArray,
+    Bool8Array,
+    UuidArray,
+    JsonArray,
+    OpaqueArray,
     scalar,
     NA,
     _NULL as NULL,
@@ -181,6 +195,8 @@ from pyarrow.lib import (
     HalfFloatScalar,
     FloatScalar,
     DoubleScalar,
+    Decimal32Scalar,
+    Decimal64Scalar,
     Decimal128Scalar,
     Decimal256Scalar,
     ListScalar,
@@ -208,6 +224,10 @@ from pyarrow.lib import (
     UnionScalar,
     RunEndEncodedScalar,
     ExtensionScalar,
+    Bool8Scalar,
+    UuidScalar,
+    JsonScalar,
+    OpaqueScalar,
 )
 
 # Buffers, allocation
@@ -372,6 +392,8 @@ __all__ = [
     "large_binary",
     "large_string",
     "large_utf8",
+    "decimal32",
+    "decimal64",
     "decimal128",
     "decimal256",
     "list_",
@@ -385,7 +407,11 @@ __all__ = [
     "dense_union",
     "dictionary",
     "run_end_encoded",
+    "json_",
+    "uuid",
     "fixed_shape_tensor",
+    "bool8",
+    "opaque",
     "field",
     "type_for_alias",
     "DataType",
@@ -413,6 +439,10 @@ __all__ = [
     "ExtensionType",
     "RunEndEncodedType",
     "FixedShapeTensorType",
+    "Bool8Type",
+    "UuidType",
+    "JsonType",
+    "OpaqueType",
     "PyExtensionType",
     "UnknownExtensionType",
     "register_extension_type",
@@ -478,6 +508,10 @@ __all__ = [
     "Decimal256Array",
     "StructArray",
     "ExtensionArray",
+    "Bool8Array",
+    "UuidArray",
+    "JsonArray",
+    "OpaqueArray",
     "RunEndEncodedArray",
     "FixedShapeTensorArray",
     "scalar",
@@ -524,6 +558,10 @@ __all__ = [
     "UnionScalar",
     "RunEndEncodedScalar",
     "ExtensionScalar",
+    "Bool8Scalar",
+    "UuidScalar",
+    "JsonScalar",
+    "OpaqueScalar",
     "DeviceAllocationType",
     "Device",
     "MemoryManager",

pyarrow-stubs/__lib_pxi/array.pyi

@@ -1680,6 +1680,10 @@ __all__ = [
     "StructArray",
     "RunEndEncodedArray",
     "ExtensionArray",
+    "Bool8Array",
+    "UuidArray",
+    "JsonArray",
+    "OpaqueArray",
     "FixedShapeTensorArray",
     "concat_arrays",
     "_empty_array",

pyarrow-stubs/__lib_pxi/scalar.pyi

@@ -256,6 +256,11 @@ class ExtensionScalar(Scalar[types.ExtensionType]):
     @staticmethod
     def from_storage(typ: types.BaseExtensionType, value) -> ExtensionScalar: ...
 
+class Bool8Scalar(Scalar[types.Bool8Type]): ...
+class UuidScalar(Scalar[types.UuidType]): ...
+class JsonScalar(Scalar[types.JsonType]): ...
+class OpaqueScalar(Scalar[types.OpaqueType]): ...
+
 class FixedShapeTensorScalar(ExtensionScalar):
     def to_numpy(self) -> np.ndarray: ...
     def to_tensor(self) -> Tensor: ...
@@ -442,6 +447,8 @@ __all__ = [
     "HalfFloatScalar",
     "FloatScalar",
     "DoubleScalar",
+    "Decimal32Scalar",
+    "Decimal64Scalar",
     "Decimal128Scalar",
     "Decimal256Scalar",
     "Date32Scalar",
@@ -470,5 +477,9 @@ __all__ = [
     "UnionScalar",
     "ExtensionScalar",
     "FixedShapeTensorScalar",
+    "Bool8Scalar",
+    "UuidScalar",
+    "JsonScalar",
+    "OpaqueScalar",
     "scalar",
 ]

pyarrow-stubs/__lib_pxi/types.pyi

@@ -694,6 +694,10 @@ __all__ = [
     "BaseExtensionType",
     "ExtensionType",
     "FixedShapeTensorType",
+    "Bool8Type",
+    "UuidType",
+    "JsonType",
+    "OpaqueType",
     "PyExtensionType",
     "UnknownExtensionType",
     "register_extension_type",