more overloads

Author: cmp0xff

pandas-stubs/core/arrays/integer.pyi

@@ -1,3 +1,4 @@
+import numpy as np
 from pandas.core.arrays.masked import BaseMaskedArray
 
 from pandas._libs.missing import NAType
@@ -16,8 +17,12 @@ class _IntegerDtype(ExtensionDtype):
 class IntegerArray(BaseMaskedArray):
     @property
     def dtype(self) -> _IntegerDtype: ...
-    def __init__(self, values, mask, copy: bool = ...) -> None: ...
-    def __setitem__(self, key, value) -> None: ...
+    def __init__(
+        self,
+        values: np.typing.NDArray[np.integer],
+        mask: np.typing.NDArray[np.bool],
+        copy: bool = False,
+    ) -> None: ...
 
 class Int8Dtype(_IntegerDtype): ...
 class Int16Dtype(_IntegerDtype): ...

pandas-stubs/core/construction.pyi

@@ -6,38 +6,59 @@ from typing import (
 )
 
 import numpy as np
+from pandas.core.arrays.arrow.array import ArrowExtensionArray
 from pandas.core.arrays.base import ExtensionArray
 from pandas.core.arrays.boolean import BooleanArray
+from pandas.core.arrays.categorical import Categorical
 from pandas.core.arrays.datetimes import DatetimeArray
 from pandas.core.arrays.floating import FloatingArray
 from pandas.core.arrays.integer import IntegerArray
+from pandas.core.arrays.interval import IntervalArray
 from pandas.core.arrays.numpy_ import NumpyExtensionArray
+from pandas.core.arrays.period import PeriodArray
+from pandas.core.arrays.sparse.array import SparseArray
+from pandas.core.arrays.string_ import StringArray
 from pandas.core.arrays.timedeltas import TimedeltaArray
 from pandas.core.indexes.base import Index
+from pandas.core.indexes.category import CategoricalIndex
 from pandas.core.indexes.datetimes import DatetimeIndex
+from pandas.core.indexes.interval import IntervalIndex
+from pandas.core.indexes.period import PeriodIndex
 from pandas.core.indexes.range import RangeIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 from pandas.core.series import Series
 
+from pandas._libs.interval import Interval
 from pandas._libs.missing import NAType
+from pandas._libs.sparse import SparseIndex
 from pandas._libs.tslibs.nattype import NaTType
+from pandas._libs.tslibs.period import Period
 from pandas._libs.tslibs.timedeltas import Timedelta
 from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import (
     BooleanDtypeArg,
+    CategoryDtypeArg,
     FloatDtypeArg,
     IntDtypeArg,
+    IntervalT,
+    SequenceNotStr,
+    StrDtypeArg,
     TimedeltaDtypeArg,
     TimestampDtypeArg,
     np_ndarray,
     np_ndarray_anyint,
     np_ndarray_bool,
     np_ndarray_dt,
     np_ndarray_float,
+    np_ndarray_str,
     np_ndarray_td,
 )
 
-from pandas.core.dtypes.dtypes import ExtensionDtype
+from pandas.core.dtypes.dtypes import (
+    ExtensionDtype,
+    IntervalDtype,
+    PeriodDtype,
+)
 
 @overload
 def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
@@ -96,12 +117,53 @@ def array(  # type: ignore[overload-overlap]
     copy: bool = True,
 ) -> TimedeltaArray: ...
 @overload
+def array(
+    data: SequenceNotStr[str | np.str_ | NAType | None] | np_ndarray_str | StringArray,
+    dtype: StrDtypeArg | None = None,
+    copy: bool = True,
+) -> StringArray: ...
+@overload
+def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+    data: Sequence[Period | NAType | None] | PeriodArray | PeriodIndex | Series[Period],
+    dtype: PeriodDtype | None = None,
+    copy: bool = True,
+) -> PeriodIndex: ...
+@overload
+def array(  # type: ignore[overload-overlap]
+    data: (
+        Sequence[IntervalT | NAType | None]
+        | IntervalIndex
+        | Series[Interval]
+        | IntervalArray
+    ),
+    dtype: IntervalDtype | None = None,
+    copy: bool = True,
+) -> IntervalArray: ...
+@overload
+def array(
+    data: Categorical | CategoricalIndex,
+    dtype: CategoryDtypeArg | None = None,
+    copy: bool = True,
+) -> Categorical: ...
+@overload
 def array(
     data: Sequence[object] | np.typing.NDArray[np.object_] | RangeIndex,
     dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
 ) -> NumpyExtensionArray: ...
 @overload
+def array(  # type: ignore[overload-overlap]
+    data: SparseArray | SparseIndex,
+    dtype: str | np.dtype | ExtensionDtype | None = None,
+    copy: bool = True,
+) -> SparseIndex: ...
+@overload
+def array(
+    data: ArrowExtensionArray,
+    dtype: str | np.dtype | ExtensionDtype | None = None,
+    copy: bool = True,
+) -> ArrowExtensionArray: ...
+@overload
 def array(
     data: Sequence[Any] | np_ndarray | ExtensionArray | Index[Any] | Series[Any],
     dtype: str | np.dtype | ExtensionDtype | None = None,