Skip to content

Commit 83997b7

Browse files
committed
take
1 parent 5431cc9 commit 83997b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pint_array/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -536,7 +536,6 @@ def nonzero(x, /):
536536

537537
def searchsorted(x1, x2, /, *, side="left", sorter=None):
538538
if sorter is not None:
539-
x1 = asarray(x1)
540539
x1 = take(x1, sorter)
541540

542541
magnitude_x1 = xp.asarray(x1.magnitude, copy=True)
@@ -820,6 +819,7 @@ def pow(x1, x2, /, *args, **kwargs):
820819

821820
## Indexing Functions
822821
def take(x, indices, /, **kwargs):
822+
x = asarray(x)
823823
magnitude = xp.take(x.magnitude, indices.magnitude, **kwargs)
824824
return ArrayUnitQuantity(magnitude, x.units)
825825

0 commit comments

Comments
 (0)