take

Author: andrewgsavage

src/pint_array/__init__.py

@@ -820,7 +820,7 @@ def pow(x1, x2, /, *args, **kwargs):
     ## Indexing Functions
     def take(x, indices, /, **kwargs):
         x = asarray(x)
-        magnitude = xp.take(x.magnitude, indices.magnitude, **kwargs)
+        magnitude = xp.take(x.magnitude, indices, **kwargs)
         return ArrayUnitQuantity(magnitude, x.units)
 
     mod.take = take