18
18
easier to adjust to future upstream changes in the analogous numpy signatures.
19
19
"""
20
20
from collections import OrderedDict
21
+ from typing import Any , Dict , Optional , Union
21
22
22
23
from numpy import ndarray
23
24
@@ -101,7 +102,8 @@ def validate_argmax_with_skipna(skipna, args, kwargs):
101
102
return skipna
102
103
103
104
104
- ARGSORT_DEFAULTS = OrderedDict ()
105
+ ARGSORT_DEFAULTS = OrderedDict () \
106
+ # type: OrderedDict[str, Optional[Union[int, str]]]
105
107
ARGSORT_DEFAULTS ['axis' ] = - 1
106
108
ARGSORT_DEFAULTS ['kind' ] = 'quicksort'
107
109
ARGSORT_DEFAULTS ['order' ] = None
@@ -110,7 +112,8 @@ def validate_argmax_with_skipna(skipna, args, kwargs):
110
112
111
113
# two different signatures of argsort, this second validation
112
114
# for when the `kind` param is supported
113
- ARGSORT_DEFAULTS_KIND = OrderedDict ()
115
+ ARGSORT_DEFAULTS_KIND = OrderedDict () \
116
+ # type: OrderedDict[str, Optional[int]]
114
117
ARGSORT_DEFAULTS_KIND ['axis' ] = - 1
115
118
ARGSORT_DEFAULTS_KIND ['order' ] = None
116
119
validate_argsort_kind = CompatValidator (ARGSORT_DEFAULTS_KIND , fname = 'argsort' ,
@@ -134,7 +137,7 @@ def validate_argsort_with_ascending(ascending, args, kwargs):
134
137
return ascending
135
138
136
139
137
- CLIP_DEFAULTS = dict (out = None )
140
+ CLIP_DEFAULTS = dict (out = None ) # type Dict[str, Any]
138
141
validate_clip = CompatValidator (CLIP_DEFAULTS , fname = 'clip' ,
139
142
method = 'both' , max_fname_arg_count = 3 )
140
143
@@ -155,13 +158,13 @@ def validate_clip_with_axis(axis, args, kwargs):
155
158
return axis
156
159
157
160
158
- COMPRESS_DEFAULTS = OrderedDict ()
161
+ COMPRESS_DEFAULTS = OrderedDict () # type: OrderedDict[str, Any]
159
162
COMPRESS_DEFAULTS ['axis' ] = None
160
163
COMPRESS_DEFAULTS ['out' ] = None
161
164
validate_compress = CompatValidator (COMPRESS_DEFAULTS , fname = 'compress' ,
162
165
method = 'both' , max_fname_arg_count = 1 )
163
166
164
- CUM_FUNC_DEFAULTS = OrderedDict ()
167
+ CUM_FUNC_DEFAULTS = OrderedDict () # type: OrderedDict[str, Any]
165
168
CUM_FUNC_DEFAULTS ['dtype' ] = None
166
169
CUM_FUNC_DEFAULTS ['out' ] = None
167
170
validate_cum_func = CompatValidator (CUM_FUNC_DEFAULTS , method = 'both' ,
@@ -185,7 +188,7 @@ def validate_cum_func_with_skipna(skipna, args, kwargs, name):
185
188
return skipna
186
189
187
190
188
- ALLANY_DEFAULTS = OrderedDict ()
191
+ ALLANY_DEFAULTS = OrderedDict () # type: OrderedDict[str, Optional[bool]]
189
192
ALLANY_DEFAULTS ['dtype' ] = None
190
193
ALLANY_DEFAULTS ['out' ] = None
191
194
ALLANY_DEFAULTS ['keepdims' ] = False
@@ -203,26 +206,27 @@ def validate_cum_func_with_skipna(skipna, args, kwargs, name):
203
206
validate_max = CompatValidator (MINMAX_DEFAULTS , fname = 'max' ,
204
207
method = 'both' , max_fname_arg_count = 1 )
205
208
206
- RESHAPE_DEFAULTS = dict (order = 'C' )
209
+ RESHAPE_DEFAULTS = dict (order = 'C' ) # type: Dict[str, str]
207
210
validate_reshape = CompatValidator (RESHAPE_DEFAULTS , fname = 'reshape' ,
208
211
method = 'both' , max_fname_arg_count = 1 )
209
212
210
- REPEAT_DEFAULTS = dict (axis = None )
213
+ REPEAT_DEFAULTS = dict (axis = None ) # type: Dict[str, Any]
211
214
validate_repeat = CompatValidator (REPEAT_DEFAULTS , fname = 'repeat' ,
212
215
method = 'both' , max_fname_arg_count = 1 )
213
216
214
- ROUND_DEFAULTS = dict (out = None )
217
+ ROUND_DEFAULTS = dict (out = None ) # type: Dict[str, Any]
215
218
validate_round = CompatValidator (ROUND_DEFAULTS , fname = 'round' ,
216
219
method = 'both' , max_fname_arg_count = 1 )
217
220
218
- SORT_DEFAULTS = OrderedDict ()
221
+ SORT_DEFAULTS = OrderedDict () \
222
+ # type: OrderedDict[str, Optional[Union[int, str]]]
219
223
SORT_DEFAULTS ['axis' ] = - 1
220
224
SORT_DEFAULTS ['kind' ] = 'quicksort'
221
225
SORT_DEFAULTS ['order' ] = None
222
226
validate_sort = CompatValidator (SORT_DEFAULTS , fname = 'sort' ,
223
227
method = 'kwargs' )
224
228
225
- STAT_FUNC_DEFAULTS = OrderedDict ()
229
+ STAT_FUNC_DEFAULTS = OrderedDict () # type: OrderedDict[str, Optional[Any]]
226
230
STAT_FUNC_DEFAULTS ['dtype' ] = None
227
231
STAT_FUNC_DEFAULTS ['out' ] = None
228
232
@@ -247,14 +251,15 @@ def validate_cum_func_with_skipna(skipna, args, kwargs, name):
247
251
validate_median = CompatValidator (MEDIAN_DEFAULTS , fname = 'median' ,
248
252
method = 'both' , max_fname_arg_count = 1 )
249
253
250
- STAT_DDOF_FUNC_DEFAULTS = OrderedDict ()
254
+ STAT_DDOF_FUNC_DEFAULTS = OrderedDict () \
255
+ # type: OrderedDict[str, Optional[bool]]
251
256
STAT_DDOF_FUNC_DEFAULTS ['dtype' ] = None
252
257
STAT_DDOF_FUNC_DEFAULTS ['out' ] = None
253
258
STAT_DDOF_FUNC_DEFAULTS ['keepdims' ] = False
254
259
validate_stat_ddof_func = CompatValidator (STAT_DDOF_FUNC_DEFAULTS ,
255
260
method = 'kwargs' )
256
261
257
- TAKE_DEFAULTS = OrderedDict ()
262
+ TAKE_DEFAULTS = OrderedDict () # type: OrderedDict[str, Optional[str]]
258
263
TAKE_DEFAULTS ['out' ] = None
259
264
TAKE_DEFAULTS ['mode' ] = 'raise'
260
265
validate_take = CompatValidator (TAKE_DEFAULTS , fname = 'take' ,
0 commit comments