diff --git a/ci/code_checks.sh b/ci/code_checks.sh index d3965c61d3f0b..a81dceffbf593 100755 --- a/ci/code_checks.sh +++ b/ci/code_checks.sh @@ -73,13 +73,9 @@ if [[ -z "$CHECK" || "$CHECK" == "docstrings" ]]; then -i "pandas.tseries.offsets.BDay PR02,SA01" \ -i "pandas.tseries.offsets.BusinessDay PR02,SA01" \ -i "pandas.tseries.offsets.BusinessHour PR02,SA01" \ - -i "pandas.tseries.offsets.CBMonthBegin PR02" \ - -i "pandas.tseries.offsets.CBMonthEnd PR02" \ -i "pandas.tseries.offsets.CDay PR02,SA01" \ -i "pandas.tseries.offsets.CustomBusinessDay PR02,SA01" \ -i "pandas.tseries.offsets.CustomBusinessHour PR02,SA01" \ - -i "pandas.tseries.offsets.CustomBusinessMonthBegin PR02" \ - -i "pandas.tseries.offsets.CustomBusinessMonthEnd PR02" \ -i "pandas.tseries.offsets.Easter.is_on_offset GL08" \ -i "pandas.tseries.offsets.FY5253.is_on_offset GL08" \ -i "pandas.tseries.offsets.FY5253Quarter.is_on_offset GL08" \ diff --git a/pandas/_libs/tslibs/offsets.pyi b/pandas/_libs/tslibs/offsets.pyi index ba850680ebbb3..27f2ee392e73b 100644 --- a/pandas/_libs/tslibs/offsets.pyi +++ b/pandas/_libs/tslibs/offsets.pyi @@ -284,8 +284,28 @@ class CustomBusinessHour(BusinessHour): offset: timedelta = ..., ) -> None: ... -class CustomBusinessMonthEnd(_CustomBusinessMonth): ... -class CustomBusinessMonthBegin(_CustomBusinessMonth): ... +class CustomBusinessMonthEnd(_CustomBusinessMonth): + def __init__( + self, + n: int = ..., + normalize: bool = ..., + weekmask: str = ..., + holidays: list = ..., + calendar: np.busdaycalendar = ..., + offset: timedelta = ..., + ) -> None: ... + +class CustomBusinessMonthBegin(_CustomBusinessMonth): + def __init__( + self, + n: int = ..., + normalize: bool = ..., + weekmask: str = ..., + holidays: list = ..., + calendar: np.busdaycalendar = ..., + offset: timedelta = ..., + ) -> None: ... + class OffsetMeta(type): ... class DateOffset(RelativeDeltaOffset, metaclass=OffsetMeta): ... diff --git a/pandas/_libs/tslibs/offsets.pyx b/pandas/_libs/tslibs/offsets.pyx index 28497a3ba2bae..74260cc2eb179 100644 --- a/pandas/_libs/tslibs/offsets.pyx +++ b/pandas/_libs/tslibs/offsets.pyx @@ -6391,6 +6391,10 @@ cdef class CustomBusinessMonthEnd(_CustomBusinessMonth): dtype='datetime64[ns]', freq='CBME') """ + _attributes = tuple( + ["n", "normalize", "weekmask", "holidays", "calendar", "offset"] + ) + _prefix = "CBME" @@ -6452,6 +6456,10 @@ cdef class CustomBusinessMonthBegin(_CustomBusinessMonth): dtype='datetime64[ns]', freq='CBMS') """ + _attributes = tuple( + ["n", "normalize", "weekmask", "holidays", "calendar", "offset"] + ) + _prefix = "CBMS"