feat: override new patterns for func repeat and nulls (zen-xu#200)

* fix: reexport decimal64 array and decimal128 array

* feat: override new patterns for func `repeat` and `nulls`

Author: zen-xu

pyarrow-stubs/__init__.pyi

@@ -168,6 +168,8 @@ from pyarrow.lib import (
     Time64Array,
     DurationArray,
     MonthDayNanoIntervalArray,
+    Decimal32Array,
+    Decimal64Array,
     Decimal128Array,
     Decimal256Array,
     StructArray,
@@ -504,6 +506,8 @@ __all__ = [
     "Time64Array",
     "DurationArray",
     "MonthDayNanoIntervalArray",
+    "Decimal32Array",
+    "Decimal64Array",
     "Decimal128Array",
     "Decimal256Array",
     "StructArray",
@@ -531,6 +535,8 @@ __all__ = [
     "HalfFloatScalar",
     "FloatScalar",
     "DoubleScalar",
+    "Decimal32Scalar",
+    "Decimal64Scalar",
     "Decimal128Scalar",
     "Decimal256Scalar",
     "ListScalar",

pyarrow-stubs/__lib_pxi/array.pyi

@@ -833,6 +833,30 @@ def nulls(
     memory_pool: MemoryPool | None = None,
 ) -> FixedShapeTensorArray: ...
 @overload
+def nulls(
+    size: int,
+    types: types.Bool8Type,
+    memory_pool: MemoryPool | None = None,
+) -> Bool8Array: ...
+@overload
+def nulls(
+    size: int,
+    types: types.UuidType,
+    memory_pool: MemoryPool | None = None,
+) -> UuidArray: ...
+@overload
+def nulls(
+    size: int,
+    types: types.JsonType,
+    memory_pool: MemoryPool | None = None,
+) -> JsonArray: ...
+@overload
+def nulls(
+    size: int,
+    types: types.OpaqueType,
+    memory_pool: MemoryPool | None = None,
+) -> OpaqueArray: ...
+@overload
 def nulls(
     size: int,
     types: types.ExtensionType,
@@ -891,6 +915,14 @@ def repeat(
     value: float | scalar.DoubleScalar, size: int, memory_pool: MemoryPool | None = None
 ) -> DoubleArray: ...
 @overload
+def repeat(
+    value: Decimal | scalar.Decimal32Scalar, size: int, memory_pool: MemoryPool | None = None
+) -> Decimal32Array: ...
+@overload
+def repeat(
+    value: Decimal | scalar.Decimal64Scalar, size: int, memory_pool: MemoryPool | None = None
+) -> Decimal64Array: ...
+@overload
 def repeat(
     value: Decimal | scalar.Decimal128Scalar, size: int, memory_pool: MemoryPool | None = None
 ) -> Decimal128Array: ...
@@ -1037,6 +1069,30 @@ def repeat(
     memory_pool: MemoryPool | None = None,
 ) -> FixedShapeTensorArray: ...
 @overload
+def repeat(
+    value: scalar.Bool8Scalar,
+    size: int,
+    memory_pool: MemoryPool | None = None,
+) -> Bool8Array: ...
+@overload
+def repeat(
+    value: scalar.UuidScalar,
+    size: int,
+    memory_pool: MemoryPool | None = None,
+) -> UuidArray: ...
+@overload
+def repeat(
+    value: scalar.JsonScalar,
+    size: int,
+    memory_pool: MemoryPool | None = None,
+) -> JsonArray: ...
+@overload
+def repeat(
+    value: scalar.OpaqueScalar,
+    size: int,
+    memory_pool: MemoryPool | None = None,
+) -> OpaqueArray: ...
+@overload
 def repeat(
     value: scalar.ExtensionScalar,
     size: int,
@@ -1660,6 +1716,8 @@ __all__ = [
     "FloatArray",
     "DoubleArray",
     "FixedSizeBinaryArray",
+    "Decimal32Array",
+    "Decimal64Array",
     "Decimal128Array",
     "Decimal256Array",
     "BaseListArray",