support build array with list of scalar (#64)

zen-xu · Sep 4, 2024 · 055ea6d · 055ea6d
1 parent 343f221
commit 055ea6d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/pyarrow-stubs/__lib_pxi/array.pyi b/pyarrow-stubs/__lib_pxi/array.pyi
@@ -1,4 +1,4 @@
-# mypy: disable-error-code="overload-overlap,misc"
+# mypy: disable-error-code="overload-overlap,misc,type-arg"
 
 import datetime as dt
 
@@ -173,6 +173,16 @@ def array(
     memory_pool: MemoryPool | None = None,
 ) -> ListArray: ...
 @overload
+def array(
+    values: NullableIterable[_Scalar_CoT],
+    type: None = None,
+    mask: Mask | None = None,
+    size: int | None = None,
+    from_pandas: bool | None = None,
+    safe: bool = True,
+    memory_pool: MemoryPool | None = None,
+) -> Array[_Scalar_CoT]: ...
+@overload
 def array(
     values: Iterable | SupportArrowArray | SupportArrowDeviceArray,
     type: _DataTypeT,

diff --git a/pyarrow-stubs/__lib_pxi/table.pyi b/pyarrow-stubs/__lib_pxi/table.pyi
@@ -202,6 +202,11 @@ def chunked_array(
     type: None = None,
 ) -> ChunkedArray[scalar.ListScalar]: ...
 @overload
+def chunked_array(
+    values: NullableIterable[_Scalar_CoT],
+    type: None = None,
+) -> ChunkedArray[_Scalar_CoT]: ...
+@overload
 def chunked_array(
     values: Iterable | SupportArrowStream | SupportArrowArray,
     type: _DataTypeT,