File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -425,8 +425,7 @@ def unique_inverse(x):
425
425
)
426
426
_manager .add_event_pair (ht_ev , sub_ev )
427
427
428
- inv_dt = dpt .int64 if x .size > dpt .iinfo (dpt .int32 ).max else dpt .int32
429
- inv = dpt .empty_like (x , dtype = inv_dt , order = "C" )
428
+ inv = dpt .empty_like (x , dtype = ind_dt , order = "C" )
430
429
ht_ev , ssl_ev = _searchsorted_left (
431
430
hay = unique_vals ,
432
431
needles = x ,
@@ -608,8 +607,7 @@ def unique_all(x: dpt.usm_ndarray) -> UniqueAllResult:
608
607
)
609
608
_manager .add_event_pair (ht_ev , sub_ev )
610
609
611
- inv_dt = dpt .int64 if x .size > dpt .iinfo (dpt .int32 ).max else dpt .int32
612
- inv = dpt .empty_like (x , dtype = inv_dt , order = "C" )
610
+ inv = dpt .empty_like (x , dtype = ind_dt , order = "C" )
613
611
ht_ev , ssl_ev = _searchsorted_left (
614
612
hay = unique_vals ,
615
613
needles = x ,
You can’t perform that action at this time.
0 commit comments