@@ -14,6 +14,7 @@ from typing import (
14
14
15
15
from dateutil .relativedelta import weekday as WeekdayClass
16
16
import numpy as np
17
+ from pandas import Timestamp
17
18
from pandas .core .indexes .datetimes import DatetimeIndex
18
19
from typing_extensions import Self
19
20
@@ -42,6 +43,10 @@ class BaseOffset:
42
43
@overload
43
44
def __add__ (self , other : npt .NDArray [np .object_ ]) -> npt .NDArray [np .object_ ]: ...
44
45
@overload
46
+ def __add__ (self , other : date ) -> Timestamp : ...
47
+ @overload
48
+ def __add__ (self , other : datetime ) -> Timestamp : ...
49
+ @overload
45
50
def __add__ (self , other : BaseOffset ) -> Self : ...
46
51
@overload
47
52
def __add__ (self , other : _DatetimeT ) -> _DatetimeT : ...
@@ -50,15 +55,28 @@ class BaseOffset:
50
55
@overload
51
56
def __radd__ (self , other : npt .NDArray [np .object_ ]) -> npt .NDArray [np .object_ ]: ...
52
57
@overload
58
+ def __radd__ (self , other : date ) -> Timestamp : ...
59
+ @overload
60
+ def __radd__ (self , other : datetime ) -> Timestamp : ...
61
+ @overload
53
62
def __radd__ (self , other : BaseOffset ) -> Self : ...
54
63
@overload
55
64
def __radd__ (self , other : _DatetimeT ) -> _DatetimeT : ...
56
65
@overload
57
66
def __radd__ (self , other : _TimedeltaT ) -> _TimedeltaT : ...
67
+ @overload
68
+ def __sub__ (self , other : date ) -> Timestamp : ...
69
+ @overload
70
+ def __sub__ (self , other : datetime ) -> Timestamp : ...
71
+ @overload
58
72
def __sub__ (self , other : BaseOffset ) -> Self : ...
59
73
@overload
60
74
def __rsub__ (self , other : npt .NDArray [np .object_ ]) -> npt .NDArray [np .object_ ]: ...
61
75
@overload
76
+ def __rsub__ (self , other : date ) -> Timestamp : ...
77
+ @overload
78
+ def __rsub__ (self , other : datetime ) -> Timestamp : ...
79
+ @overload
62
80
def __rsub__ (self , other : BaseOffset ) -> Self : ...
63
81
@overload
64
82
def __rsub__ (self , other : _DatetimeT ) -> _DatetimeT : ...
0 commit comments