File tree 2 files changed +7
-2
lines changed
2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -2896,7 +2896,7 @@ def _get_cythonized_result(
2896
2896
grouper = self .grouper
2897
2897
2898
2898
ids , _ , ngroups = grouper .group_info
2899
- output : dict [base .OutputKey , np . ndarray ] = {}
2899
+ output : dict [base .OutputKey , ArrayLike ] = {}
2900
2900
2901
2901
base_func = getattr (libgroupby , how )
2902
2902
base_func = partial (base_func , labels = ids )
@@ -2911,6 +2911,7 @@ def blk_func(values: ArrayLike) -> ArrayLike:
2911
2911
else :
2912
2912
result_sz = len (values )
2913
2913
2914
+ result : ArrayLike
2914
2915
result = np .zeros (result_sz , dtype = cython_dtype )
2915
2916
if needs_2d :
2916
2917
result = result .reshape ((- 1 , 1 ))
Original file line number Diff line number Diff line change @@ -418,7 +418,11 @@ def _bins_to_cuts(
418
418
bins = unique_bins
419
419
420
420
side = "left" if right else "right"
421
- ids = ensure_platform_int (bins .searchsorted (x , side = side ))
421
+ # error: No overload variant of "searchsorted" of "ndarray" matches
422
+ # argument types "Any", "str"
423
+ ids = ensure_platform_int (
424
+ bins .searchsorted (x , side = side ) # type: ignore[call-overload]
425
+ )
422
426
423
427
if include_lowest :
424
428
ids [np .asarray (x ) == bins [0 ]] = 1
You can’t perform that action at this time.
0 commit comments