Skip to content

Commit 46eea6f

Browse files
committed
fix: pyrefly
1 parent 6100ccf commit 46eea6f

File tree

3 files changed

+16
-3
lines changed

3 files changed

+16
-3
lines changed

pandas-stubs/core/indexes/accessors.pyi

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ from typing_extensions import Never
3939
from pandas._libs.interval import Interval
4040
from pandas._libs.tslibs import BaseOffset
4141
from pandas._libs.tslibs.offsets import DateOffset
42+
from pandas._libs.tslibs.period import Period
4243
from pandas._libs.tslibs.timedeltas import Timedelta
4344
from pandas._libs.tslibs.timestamps import Timestamp
4445
from pandas._typing import (
@@ -456,6 +457,12 @@ class DtDescriptor:
456457
def __get__(
457458
self, instance: Series[Timedelta], owner: type[Series]
458459
) -> TimedeltaProperties: ...
460+
@overload
461+
def __get__(
462+
self,
463+
instance: Series[Period] | PeriodSeries,
464+
owner: type[Series | PeriodSeries],
465+
) -> PeriodProperties: ...
459466

460467
@type_check_only
461468
class ArrayDescriptor:

pandas-stubs/core/series.pyi

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ from pandas.core.groupby.groupby import BaseGroupBy
7070
from pandas.core.indexers import BaseIndexer
7171
from pandas.core.indexes.accessors import (
7272
DtDescriptor,
73-
PeriodProperties,
7473
)
7574
from pandas.core.indexes.category import CategoricalIndex
7675
from pandas.core.indexes.datetimes import DatetimeIndex
@@ -4724,8 +4723,7 @@ class _SeriesSubclassBase(Series[S1], Generic[S1, GenericT_co]):
47244723
) -> np_1darray: ...
47254724

47264725
class PeriodSeries(_SeriesSubclassBase[Period, np.object_]):
4727-
@property
4728-
def dt(self) -> PeriodProperties: ...
4726+
dt: ClassVar = DtDescriptor() # noqa
47294727
def __sub__(self, other: PeriodSeries) -> OffsetSeries: ... # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
47304728
def diff(self, periods: int = ...) -> OffsetSeries: ... # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
47314729

tests/series/test_properties.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from pandas.core.arrays.timedeltas import TimedeltaArray
1212
from pandas.core.indexes.accessors import (
1313
DatetimeProperties,
14+
PeriodProperties,
1415
TimedeltaProperties,
1516
)
1617
from typing_extensions import assert_type
@@ -34,6 +35,13 @@ def test_dt_property() -> None:
3435
assert_type(pd.Series([pd.Timedelta(1, "s")]).dt, TimedeltaProperties),
3536
TimedeltaProperties,
3637
)
38+
check(
39+
assert_type(
40+
pd.period_range(start="2022-06-01", periods=10).to_series().dt,
41+
PeriodProperties,
42+
),
43+
PeriodProperties,
44+
)
3745

3846
if TYPE_CHECKING_INVALID_USAGE:
3947
assert_type(pd.DataFrame({"a": [1]})["a"].dt, Any)

0 commit comments

Comments
 (0)