Merge branch 'main' of github.com:loicdiridollou/pandas-stubs into gh1089_fix_nightly

Author: loicdiridollou

README.md

@@ -9,6 +9,7 @@
 [![Powered by NumFOCUS](https://img.shields.io/badge/powered%20by-NumFOCUS-orange.svg?style=flat&colorA=E1523D&colorB=007D8A)](https://numfocus.org)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)
+[![Typing: stubs](https://img.shields.io/pypi/types/pandas-stubs)](https://pypi.org/project/pandas-stubs/)
 
 ## What is it?
 

pandas-stubs/_libs/tslibs/dtypes.pyi

@@ -1,3 +1,4 @@
+# pyright: strict
 from datetime import (
     datetime,
     timedelta,
@@ -11,7 +12,10 @@ from typing_extensions import (
 )
 
 from pandas._libs.tslibs.period import Period
-from pandas._typing import TimeUnit
+from pandas._typing import (
+    NpDtype,
+    TimeUnit,
+)
 
 NaT: NaTType
 iNaT: int
@@ -30,7 +34,7 @@ class NaTType:
     def asm8(self) -> np.datetime64: ...
     def to_datetime64(self) -> np.datetime64: ...
     def to_numpy(
-        self, dtype: np.dtype | str | None = ..., copy: bool = ...
+        self, dtype: NpDtype | None = ..., copy: bool = ...
     ) -> np.datetime64 | np.timedelta64: ...
     @property
     def is_leap_year(self) -> bool: ...
@@ -67,8 +71,8 @@ class NaTType:
     def weekday(self) -> float: ...
     def isoweekday(self) -> float: ...
     def total_seconds(self) -> float: ...
-    def today(self, *args, **kwargs) -> NaTType: ...
-    def now(self, *args, **kwargs) -> NaTType: ...
+    def today(self, tz: _tzinfo | str | None = ...) -> NaTType: ...
+    def now(self, tz: _tzinfo | str | None = ...) -> NaTType: ...
     def to_pydatetime(self) -> NaTType: ...
     def date(self) -> NaTType: ...
     def round(self) -> NaTType: ...

pandas-stubs/_libs/tslibs/nattype.pyi

@@ -9,10 +9,7 @@ from pandas.core.arrays.datetimes import DatetimeArray as DatetimeArray
 from pandas.core.arrays.integer import IntegerArray as IntegerArray
 from pandas.core.arrays.interval import IntervalArray as IntervalArray
 from pandas.core.arrays.numpy_ import PandasArray as PandasArray
-from pandas.core.arrays.period import (
-    PeriodArray as PeriodArray,
-    period_array as period_array,
-)
+from pandas.core.arrays.period import PeriodArray as PeriodArray
 from pandas.core.arrays.sparse import SparseArray as SparseArray
 from pandas.core.arrays.string_ import StringArray as StringArray
 from pandas.core.arrays.timedeltas import TimedeltaArray as TimedeltaArray

pandas-stubs/_libs/tslibs/tzconversion.pyi

@@ -1,5 +1,3 @@
-from collections.abc import Sequence
-
 import numpy as np
 from pandas import PeriodDtype
 from pandas.core.arrays.datetimelike import (
@@ -10,8 +8,6 @@ from pandas.core.arrays.datetimelike import (
 from pandas._libs.tslibs import Timestamp
 from pandas._libs.tslibs.period import Period
 
-from pandas.tseries.offsets import Tick
-
 class PeriodArray(DatetimeLikeArrayMixin, DatelikeOps):
     __array_priority__: int = ...
     def __init__(self, values, freq=..., dtype=..., copy: bool = ...) -> None: ...
@@ -44,12 +40,3 @@ class PeriodArray(DatetimeLikeArrayMixin, DatelikeOps):
     def to_timestamp(self, freq: str | None = ..., how: str = ...) -> Timestamp: ...
     def asfreq(self, freq: str | None = ..., how: str = ...) -> Period: ...
     def astype(self, dtype, copy: bool = ...): ...
-
-def raise_on_incompatible(left, right): ...
-def period_array(
-    data: Sequence[Period | None],
-    freq: str | Tick | None = ...,
-    copy: bool = ...,
-) -> PeriodArray: ...
-def validate_dtype_freq(dtype, freq): ...
-def dt64arr_to_periodarr(data, freq, tz=...): ...

pandas-stubs/_libs/tslibs/vectorized.pyi

@@ -13,6 +13,7 @@ from typing import (
     Any,
     ClassVar,
     Literal,
+    TypeAlias,
     final,
     overload,
 )
@@ -263,7 +264,16 @@ class Index(IndexOpsMixin[S1]):
     @property
     def str(
         self,
-    ) -> StringMethods[Self, MultiIndex, np_ndarray_bool, Index[list[str]]]: ...
+    ) -> StringMethods[
+        Self,
+        MultiIndex,
+        np_ndarray_bool,
+        Index[list[str]],
+        Index[int],
+        Index[bytes],
+        Index[str],
+        Index[type[object]],
+    ]: ...
     def is_(self, other) -> bool: ...
     def __len__(self) -> int: ...
     def __array__(self, dtype=...) -> np.ndarray: ...
@@ -455,6 +465,8 @@ class Index(IndexOpsMixin[S1]):
         ),
     ) -> Self: ...
 
+UnknownIndex: TypeAlias = Index[Any]
+
 def ensure_index_from_sequences(
     sequences: Sequence[Sequence[Dtype]], names: list[str] = ...
 ) -> Index: ...

pandas-stubs/core/arrays/__init__.pyi

@@ -1179,7 +1179,16 @@ class Series(IndexOpsMixin[S1], NDFrame):
     @property
     def str(
         self,
-    ) -> StringMethods[Series, DataFrame, Series[bool], Series[list[str]]]: ...
+    ) -> StringMethods[
+        Self,
+        DataFrame,
+        Series[bool],
+        Series[list[str]],
+        Series[int],
+        Series[bytes],
+        Series[str],
+        Series[type[object]],
+    ]: ...
     @property
     def dt(self) -> CombinedDatetimelikeProperties: ...
     @property
@@ -2318,3 +2327,5 @@ class IntervalSeries(Series[Interval[_OrderableT]], Generic[_OrderableT]):
     @property
     def array(self) -> IntervalArray: ...
     def diff(self, periods: int = ...) -> Never: ...
+
+UnknownSeries: TypeAlias = Series[Any]