Fixed formatting issues

Author: msfterictraut

pandas/_libs/interval.pyi

@@ -14,7 +14,6 @@ iNaT: int
 nat_strings: set[str]
 
 def is_null_datetimelike(val: object, inat_is_null: bool = ...) -> bool: ...
-def checknull_with_nat(val: object) -> bool: ...
 
 class NaTType(datetime):
     value: np.int64

pandas/_libs/tslibs/nattype.pyi

@@ -1,9 +1,39 @@
 from __future__ import annotations
-from datetime import datetime
-from typing import Any, Tuple, Union
-from datetime import timedelta
+
+from datetime import (
+    datetime,
+    timedelta,
+)
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Collection,
+    Literal,
+    Tuple,
+    TypeVar,
+    Union,
+    overload,
+)
+
+import numpy as np
+
+from pandas._typing import npt
+
+from .timedeltas import Timedelta
+
+if TYPE_CHECKING:
+    from pandas.core.indexes.datetimes import DatetimeIndex
+_BaseOffsetT = TypeVar("_BaseOffsetT", bound="BaseOffset")
+_DatetimeT = TypeVar("_DatetimeT", bound=datetime)
+_TimedeltaT = TypeVar("_TimedeltaT", bound=timedelta)
+
+_relativedelta_kwds: set[str]
+prefix_mapping: dict[str, type]
+
+class ApplyTypeError(TypeError): ...
 
 class BaseOffset:
+    n: int
     def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
     def __eq__(self, other) -> bool: ...
     def __ne__(self, other) -> bool: ...
@@ -12,21 +42,49 @@ class BaseOffset:
     def kwds(self) -> dict: ...
     @property
     def base(self) -> BaseOffset: ...
-    def __add__(self, other) -> BaseOffset: ...
-    def __sub__(self, other) -> BaseOffset: ...
+    @overload
+    def __add__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ...
+    @overload
+    def __add__(self: _BaseOffsetT, other: BaseOffset) -> _BaseOffsetT: ...
+    @overload
+    def __add__(self, other: _DatetimeT) -> _DatetimeT: ...
+    @overload
+    def __add__(self, other: _TimedeltaT) -> _TimedeltaT: ...
+    @overload
+    def __radd__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ...
+    @overload
+    def __radd__(self: _BaseOffsetT, other: BaseOffset) -> _BaseOffsetT: ...
+    @overload
+    def __radd__(self, other: _DatetimeT) -> _DatetimeT: ...
+    @overload
+    def __radd__(self, other: _TimedeltaT) -> _TimedeltaT: ...
+    def __sub__(self: _BaseOffsetT, other: BaseOffset) -> _BaseOffsetT: ...
+    @overload
+    def __rsub__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ...
+    @overload
+    def __rsub__(self: _BaseOffsetT, other: BaseOffset) -> _BaseOffsetT: ...
+    @overload
+    def __rsub__(self, other: _DatetimeT) -> _DatetimeT: ...
+    @overload
+    def __rsub__(self, other: _TimedeltaT) -> _TimedeltaT: ...
     def __call__(self, other): ...
-    def __mul__(self, other): ...
-    def __neg__(self) -> BaseOffset: ...
-    def copy(self) -> BaseOffset: ...
+    @overload
+    def __mul__(self, other: np.ndarray) -> np.ndarray: ...
+    @overload
+    def __mul__(self: _BaseOffsetT, other: int) -> _BaseOffsetT: ...
+    @overload
+    def __rmul__(self, other: np.ndarray) -> np.ndarray: ...
+    @overload
+    def __rmul__(self: _BaseOffsetT, other: int) -> _BaseOffsetT: ...
+    def __neg__(self: _BaseOffsetT) -> _BaseOffsetT: ...
+    def copy(self: _BaseOffsetT) -> _BaseOffsetT: ...
     def __repr__(self) -> str: ...
     @property
     def name(self) -> str: ...
     @property
     def rule_code(self) -> str: ...
     def freqstr(self) -> str: ...
-    # Next one is problematic due to circular imports
-    #def apply_index(self, dtindex: DatetimeIndex) -> DatetimeIndex: ...
-    def apply_index(self, dtindex): ...
+    def apply_index(self, dtindex: "DatetimeIndex") -> "DatetimeIndex": ...
     def _apply_array(self, dtarr) -> None: ...
     def rollback(self, dt: datetime) -> datetime: ...
     def rollforward(self, dt: datetime) -> datetime: ...
@@ -39,15 +97,26 @@ class BaseOffset:
     def isAnchored(self) -> bool: ...
     def is_anchored(self) -> bool: ...
 
+def _get_offset(name: str) -> BaseOffset: ...
+
 class SingleConstructorOffset(BaseOffset):
     @classmethod
-    def _from_name(cls, suffix=None): ...
+    def _from_name(cls, suffix=...): ...
     def __reduce__(self): ...
 
-def to_offset(freq: Union[str, Tuple, timedelta, BaseOffset, None]) -> Union[BaseOffset, None]: ...
+@overload
+def to_offset(freq: None) -> None: ...
+@overload
+def to_offset(freq: timedelta | BaseOffset | str) -> BaseOffset: ...
 
 class Tick(SingleConstructorOffset):
     def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
+    @property
+    def delta(self) -> Timedelta: ...
+    @property
+    def nanos(self) -> int: ...
+
+def delta_to_tick(delta: timedelta) -> Tick: ...
 
 class Day(Tick): ...
 class Hour(Tick): ...
@@ -56,18 +125,44 @@ class Second(Tick): ...
 class Milli(Tick): ...
 class Micro(Tick): ...
 class Nano(Tick): ...
+
 class RelativeDeltaOffset(BaseOffset):
     def __init__(self, n: int = ..., normalize: bool = ..., **kwds: Any) -> None: ...
-class BusinessMixin(SingleConstructorOffset): ...
+
+class BusinessMixin(SingleConstructorOffset):
+    def __init__(
+        self, n: int = ..., normalize: bool = ..., offset: timedelta = ...
+    ): ...
+
 class BusinessDay(BusinessMixin): ...
-class BusinessHour(BusinessMixin): ...
+
+class BusinessHour(BusinessMixin):
+    def __init__(
+        self,
+        n: int = ...,
+        normalize: bool = ...,
+        start: str | Collection[str] = ...,
+        end: str | Collection[str] = ...,
+        offset: timedelta = ...,
+    ): ...
+
 class WeekOfMonthMixin(SingleConstructorOffset): ...
-class YearOffset(SingleConstructorOffset): ...
+
+class YearOffset(SingleConstructorOffset):
+    def __init__(
+        self, n: int = ..., normalize: bool = ..., month: int | None = ...
+    ): ...
+
 class BYearEnd(YearOffset): ...
 class BYearBegin(YearOffset): ...
 class YearEnd(YearOffset): ...
 class YearBegin(YearOffset): ...
-class QuarterOffset(SingleConstructorOffset): ...
+
+class QuarterOffset(SingleConstructorOffset):
+    def __init__(
+        self, n: int = ..., normalize: bool = ..., startingMonth: int | None = ...
+    ) -> None: ...
+
 class BQuarterEnd(QuarterOffset): ...
 class BQuarterBegin(QuarterOffset): ...
 class QuarterEnd(QuarterOffset): ...
@@ -77,19 +172,66 @@ class MonthEnd(MonthOffset): ...
 class MonthBegin(MonthOffset): ...
 class BusinessMonthEnd(MonthOffset): ...
 class BusinessMonthBegin(MonthOffset): ...
-class SemiMonthOffset(SingleConstructorOffset): ...
+
+class SemiMonthOffset(SingleConstructorOffset):
+    def __init__(
+        self, n: int = ..., normalize: bool = ..., day_of_month: int | None = ...
+    ) -> None: ...
+
 class SemiMonthEnd(SemiMonthOffset): ...
 class SemiMonthBegin(SemiMonthOffset): ...
-class Week(SingleConstructorOffset): ...
+
+class Week(SingleConstructorOffset):
+    def __init__(
+        self, n: int = ..., normalize: bool = ..., weekday: int | None = ...
+    ) -> None: ...
+
 class WeekOfMonth(WeekOfMonthMixin): ...
 class LastWeekOfMonth(WeekOfMonthMixin): ...
-class FY5253Mixin(SingleConstructorOffset): ...
+
+class FY5253Mixin(SingleConstructorOffset):
+    def __init__(
+        self,
+        n: int = ...,
+        normalize: bool = ...,
+        weekday: int = ...,
+        startingMonth: int = ...,
+        variation: str = ...,
+    ) -> None: ...
+
 class FY5253(FY5253Mixin): ...
 class FY5253Quarter(FY5253Mixin): ...
 class Easter(SingleConstructorOffset): ...
-class _CustomBusinessMonth(BusinessMixin): ...
-class CustomBusinessDay(BusinessDay): ...
-class CustomBusinessHour(BusinessHour): ...
+
+class _CustomBusinessMonth(BusinessMixin):
+    def __init__(
+        self,
+        n: int = ...,
+        normalize: bool = ...,
+        offset: timedelta = ...,
+        holidays: None | list = ...,
+    ): ...
+
+class CustomBusinessDay(BusinessDay):
+    def __init__(
+        self,
+        n: int = ...,
+        normalize: bool = ...,
+        offset: timedelta = ...,
+        weekmask: str = ...,
+    ): ...
+
+class CustomBusinessHour(BusinessHour):
+    def __init__(
+        self,
+        n: int = ...,
+        normalize: bool = ...,
+        start: str = ...,
+        end: str = ...,
+        offset: timedelta = ...,
+        holidays: None | list = ...,
+    ): ...
+
 class CustomBusinessMonthEnd(_CustomBusinessMonth): ...
 class CustomBusinessMonthBegin(_CustomBusinessMonth): ...
 class DateOffset(RelativeDeltaOffset): ...
@@ -100,3 +242,15 @@ BMonthBegin = BusinessMonthBegin
 CBMonthEnd = CustomBusinessMonthEnd
 CBMonthBegin = CustomBusinessMonthBegin
 CDay = CustomBusinessDay
+
+def roll_qtrday(
+    other: datetime, n: int, month: int, day_opt: str, modby: int
+) -> int: ...
+
+INVALID_FREQ_ERR_MSG: Literal["Invalid frequency: {0}"]
+
+def shift_months(
+    dtindex: npt.NDArray[np.int64], months: int, day_opt: str | None = ...
+) -> npt.NDArray[np.int64]: ...
+
+_offset_map: dict[str, BaseOffset]

pandas/_libs/tslibs/offsets.pyi

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import timedelta
 from typing import Literal
 
 import numpy as np
@@ -15,9 +15,6 @@ from pandas._typing import (
 INVALID_FREQ_ERR_MSG: str
 DIFFERENT_FREQ: str
 
-def is_period_object(obj: object) -> bool: ...
-def get_period_ordinal(dts: datetime, freq: int) -> int: ...
-
 class IncompatibleFrequency(ValueError): ...
 
 def periodarr_to_dt64arr(
@@ -37,7 +34,7 @@ def get_period_field_arr(
 ) -> npt.NDArray[np.int64]: ...
 def from_ordinals(
     values: npt.NDArray[np.int64],  # const int64_t[:]
-    freq: Frequency,
+    freq: timedelta | BaseOffset | str | None,
 ) -> npt.NDArray[np.int64]: ...
 def extract_ordinals(
     values: npt.NDArray[np.object_],
@@ -63,7 +60,7 @@ class Period:
     def __new__(  # type: ignore[misc]
         cls,
         value=...,
-        freq: int | str | None = ...,
+        freq: int | str | BaseOffset | None = ...,
         ordinal: int | None = ...,
         year: int | None = ...,
         month: int | None = ...,
@@ -86,7 +83,7 @@ class Period:
         how: str = ...,
         tz: Timezone | None = ...,
     ) -> Timestamp: ...
-    def asfreq(self, freq: str, how: str = ...) -> Period: ...
+    def asfreq(self, freq: str | BaseOffset, how: str = ...) -> Period: ...
     @property
     def freqstr(self) -> str: ...
     @property

pandas/_libs/tslibs/period.pyi

@@ -1795,7 +1795,7 @@ cdef class _Period(PeriodMixin):
 
         Parameters
         ----------
-        freq : str
+        freq : str, BaseOffset
             The desired frequency.
         how : {'E', 'S', 'end', 'start'}, default 'end'
             Start or end of the timespan.

pandas/_libs/tslibs/period.pyx

@@ -67,10 +67,19 @@ class Timedelta(timedelta):
     def __abs__(self) -> timedelta: ...
     def __mul__(self, other: float) -> timedelta: ...
     def __rmul__(self, other: float) -> timedelta: ...
-    @overload
+    # error: Signature of "__floordiv__" incompatible with supertype "timedelta"
+    @overload  # type: ignore[override]
     def __floordiv__(self, other: timedelta) -> int: ...
     @overload
-    def __floordiv__(self, other: int) -> timedelta: ...
+    def __floordiv__(self, other: int | float) -> timedelta: ...
+    @overload
+    def __floordiv__(self, other: np.ndarray) -> npt.NDArray[np.timedelta64]: ...
+    @overload
+    def __rfloordiv__(self, other: timedelta) -> int: ...
+    @overload
+    def __rfloordiv__(self, other: int | float) -> timedelta: ...
+    @overload
+    def __rfloordiv__(self, other: np.ndarray) -> npt.NDArray[np.timedelta64]: ...
     @overload
     def __truediv__(self, other: timedelta) -> float: ...
     @overload