13
13
from pandas .core .arrays .interval import VALID_CLOSED
14
14
15
15
16
- def fallback_performancewarning (version : str | None = None ):
16
+ def fallback_performancewarning (version : str | None = None ) -> None :
17
17
"""
18
18
Raise a PerformanceWarning for falling back to ExtensionArray's
19
19
non-pyarrow method
@@ -24,7 +24,9 @@ def fallback_performancewarning(version: str | None = None):
24
24
warnings .warn (msg , PerformanceWarning , stacklevel = find_stack_level ())
25
25
26
26
27
- def pyarrow_array_to_numpy_and_mask (arr , dtype : np .dtype ):
27
+ def pyarrow_array_to_numpy_and_mask (
28
+ arr , dtype : np .dtype
29
+ ) -> tuple [np .ndarray , np .ndarray ]:
28
30
"""
29
31
Convert a primitive pyarrow.Array to a numpy array and boolean mask based
30
32
on the buffers of the Array.
@@ -74,12 +76,12 @@ def __init__(self, freq) -> None:
74
76
def freq (self ):
75
77
return self ._freq
76
78
77
- def __arrow_ext_serialize__ (self ):
79
+ def __arrow_ext_serialize__ (self ) -> bytes :
78
80
metadata = {"freq" : self .freq }
79
81
return json .dumps (metadata ).encode ()
80
82
81
83
@classmethod
82
- def __arrow_ext_deserialize__ (cls , storage_type , serialized ):
84
+ def __arrow_ext_deserialize__ (cls , storage_type , serialized ) -> ArrowPeriodType :
83
85
metadata = json .loads (serialized .decode ())
84
86
return ArrowPeriodType (metadata ["freq" ])
85
87
@@ -122,7 +124,7 @@ def subtype(self):
122
124
return self ._subtype
123
125
124
126
@property
125
- def inclusive (self ):
127
+ def inclusive (self ) -> str :
126
128
return self ._closed
127
129
128
130
@property
@@ -134,12 +136,12 @@ def closed(self):
134
136
)
135
137
return self ._closed
136
138
137
- def __arrow_ext_serialize__ (self ):
139
+ def __arrow_ext_serialize__ (self ) -> bytes :
138
140
metadata = {"subtype" : str (self .subtype ), "inclusive" : self .inclusive }
139
141
return json .dumps (metadata ).encode ()
140
142
141
143
@classmethod
142
- def __arrow_ext_deserialize__ (cls , storage_type , serialized ):
144
+ def __arrow_ext_deserialize__ (cls , storage_type , serialized ) -> ArrowIntervalType :
143
145
metadata = json .loads (serialized .decode ())
144
146
subtype = pyarrow .type_for_alias (metadata ["subtype" ])
145
147
inclusive = metadata ["inclusive" ]
0 commit comments