Merge remote-tracking branch 'upstream/main' into scalars-interval

Author: bashtage

.pre-commit-config.yaml

@@ -11,7 +11,7 @@ repos:
     hooks:
     -   id: isort
 -   repo: https://github.com/asottile/pyupgrade
-    rev: v2.37.3
+    rev: v3.2.0
     hooks:
     -   id: pyupgrade
         types_or: [python, pyi]
@@ -27,7 +27,7 @@ repos:
     -   id: flake8
         name: flake8 (pyi)
         additional_dependencies:
-        - flake8-pyi==22.8.2
+        - flake8-pyi==22.10.0
         types: [pyi]
         args: [
           --ignore=E301 E302 E305 E402 E501 E701 E704 F401 F811 W503 Y019 Y027 Y034 Y037 Y041 Y042,

pandas-stubs/_libs/tslibs/offsets.pyi

@@ -1,6 +1,7 @@
 from datetime import (
     date,
     datetime,
+    time,
     timedelta,
 )
 from typing import (
@@ -78,6 +79,7 @@ class BaseOffset:
     def name(self) -> str: ...
     @property
     def rule_code(self) -> str: ...
+    @property
     def freqstr(self) -> str: ...
     def apply_index(self, dtindex: DatetimeIndex) -> DatetimeIndex: ...
     def rollback(self, dt: datetime) -> datetime: ...
@@ -131,8 +133,8 @@ class BusinessHour(BusinessMixin):
         self,
         n: int = ...,
         normalize: bool = ...,
-        start: str | Collection[str] = ...,
-        end: str | Collection[str] = ...,
+        start: str | time | Collection[str | time] = ...,
+        end: str | time | Collection[str | time] = ...,
         offset: timedelta = ...,
     ): ...
 
@@ -217,8 +219,8 @@ class CustomBusinessHour(BusinessHour):
         self,
         n: int = ...,
         normalize: bool = ...,
-        start: str = ...,
-        end: str = ...,
+        start: str | time | Collection[str | time] = ...,
+        end: str | time | Collection[str | time] = ...,
         offset: timedelta = ...,
         holidays: list | None = ...,
     ): ...

pandas-stubs/_testing/__init__.pyi

@@ -74,7 +74,7 @@ def assert_series_equal(
     obj: str = ...,
     *,
     check_index: Literal[False],
-    check_like: Literal[False],
+    check_like: Literal[False] = ...,
 ) -> None: ...
 @overload
 def assert_series_equal(

pandas-stubs/core/base.pyi

@@ -63,6 +63,7 @@ class IndexOpsMixin:
     def tolist(self) -> list: ...
     def to_list(self) -> list: ...
     def __iter__(self): ...
+    @property
     def hasnans(self) -> bool: ...
     def value_counts(
         self,

pandas-stubs/core/dtypes/common.pyi

@@ -28,7 +28,9 @@ from pandas.core.dtypes.inference import (
     is_scalar as is_scalar,
 )
 
-_ArrayOrDtype: TypeAlias = Union[ArrayLike, npt.DTypeLike, pd.Series, pd.DataFrame]
+_ArrayOrDtype: TypeAlias = Union[
+    ArrayLike, npt.DTypeLike, pd.Series, pd.DataFrame, pd.Index
+]
 
 def is_object_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ...
 def is_sparse(arr: ArrayLike | pd.Series | pd.DataFrame) -> bool: ...

pandas-stubs/core/frame.pyi

@@ -1276,8 +1276,6 @@ class DataFrame(NDFrame, OpsMixin):
     @property
     def at(self): ...  # Not sure what to do with this yet; look at source
     @property
-    def bool(self) -> _bool: ...
-    @property
     def columns(self) -> Index: ...
     @columns.setter  # setter needs to be right next to getter; otherwise mypy complains
     def columns(
@@ -1359,7 +1357,7 @@ class DataFrame(NDFrame, OpsMixin):
     ) -> DataFrame: ...
     def astype(
         self,
-        dtype: _str | Dtype | dict[_str, _str | Dtype],
+        dtype: _str | Dtype | dict[_str, _str | Dtype] | Series,
         copy: _bool = ...,
         errors: _str = ...,
     ) -> DataFrame: ...

pandas-stubs/core/generic.pyi

@@ -8,6 +8,7 @@ from typing import (
     Literal,
     Mapping,
     Sequence,
+    final,
     overload,
 )
 
@@ -83,6 +84,7 @@ class NDFrame(PandasObject, indexing.IndexingMixin):
     def __neg__(self: NDFrameT) -> NDFrameT: ...
     def __pos__(self: NDFrameT) -> NDFrameT: ...
     def __nonzero__(self) -> None: ...
+    @final
     def bool(self) -> _bool: ...
     def __abs__(self) -> NDFrame: ...
     def __round__(self, decimals: int = ...) -> NDFrame: ...

pandas-stubs/core/groupby/ops.pyi

@@ -70,6 +70,7 @@ class BinGrouper(BaseGrouper):
     def get_iterator(self, data: DataFrame | Series, axis: int = ...): ...
     def indices(self): ...
     def group_info(self): ...
+    @property
     def reconstructed_codes(self) -> list[np.ndarray]: ...
     def result_index(self): ...
     @property

pandas-stubs/core/indexes/base.pyi

@@ -112,6 +112,7 @@ class Index(IndexOpsMixin, PandasObject):
     def is_monotonic_increasing(self) -> bool: ...
     @property
     def is_monotonic_decreasing(self) -> bool: ...
+    @property
     def is_unique(self) -> bool: ...
     @property
     def has_duplicates(self) -> bool: ...
@@ -124,9 +125,10 @@ class Index(IndexOpsMixin, PandasObject):
     def is_interval(self) -> bool: ...
     def is_mixed(self) -> bool: ...
     def holds_integer(self): ...
+    @property
     def inferred_type(self): ...
-    def is_all_dates(self) -> bool: ...
     def __reduce__(self): ...
+    @property
     def hasnans(self) -> bool: ...
     def isna(self): ...
     isnull = ...
@@ -177,6 +179,7 @@ class Index(IndexOpsMixin, PandasObject):
     ): ...
     @property
     def values(self) -> np.ndarray: ...
+    @property
     def array(self) -> ExtensionArray: ...
     def memory_usage(self, deep: bool = ...): ...
     def where(self, cond, other=...): ...

pandas-stubs/core/indexes/category.pyi

@@ -28,6 +28,7 @@ class CategoricalIndex(ExtensionIndex, accessor.PandasDelegate):
     def __array__(self, dtype=...) -> np.ndarray: ...
     def astype(self, dtype: DtypeArg, copy: bool = ...) -> Index: ...
     def fillna(self, value=..., downcast=...): ...
+    @property
     def is_unique(self) -> bool: ...
     @property
     def is_monotonic_increasing(self) -> bool: ...