support pyarrow compute funcs

pyarrow-stubs/__lib_pxi/scalar.pyi

@@ -1,4 +1,4 @@
-# mypy: disable-error-code="overload-overlap,misc"
+# mypy: disable-error-code="overload-overlap,misc,type-arg"
 import collections.abc
 import datetime as dt
 
@@ -281,6 +281,10 @@ def scalar(
     value: Decimal, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
 ) -> Decimal128Scalar: ...
 @overload
+def scalar(
+    value: dt.datetime, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
+) -> TimestampScalar: ...
+@overload
 def scalar(
     value: dt.date, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
 ) -> Date32Scalar: ...
@@ -349,6 +353,13 @@ def scalar(
     memory_pool: MemoryPool | None = None,
 ) -> ListScalar[types.ListType[types.Decimal128Type]]: ...
 @overload
+def scalar(
+    value: CollectionValue[dt.datetime],
+    *,
+    from_pandas: bool | None = None,
+    memory_pool: MemoryPool | None = None,
+) -> ListScalar[types.ListType[types.TimestampType]]: ...
+@overload
 def scalar(
     value: CollectionValue[dt.date],
     *,

pyarrow-stubs/__lib_pxi/table.pyi

@@ -1,4 +1,4 @@
-# mypy: disable-error-code="overload-overlap"
+# mypy: disable-error-code="overload-overlap,type-arg,misc"
 
 import datetime as dt
 
@@ -162,6 +162,11 @@ def chunked_array(
     type: None = None,
 ) -> ChunkedArray[scalar.StructScalar]: ...
 @overload
+def chunked_array(
+    values: NullableIterable[dt.datetime],
+    type: None = None,
+) -> ChunkedArray[scalar.TimestampScalar]: ...
+@overload
 def chunked_array(
     values: NullableIterable[dt.date],
     type: None = None,