Update _indexing_functions.py

pydata · Aug 25, 2024 · 45abdfd · 45abdfd
1 parent 3cee797
commit 45abdfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xarray/namedarray/_array_api/_indexing_functions.py b/xarray/namedarray/_array_api/_indexing_functions.py
@@ -18,7 +18,7 @@ def take(
     axis: int | None = None,
 ) -> NamedArray:
     xp = _get_data_namespace(x)
-    _axis = _dims_to_axis(x, dim, axis)
+    _axis = _dims_to_axis(x, dim, axis)[0]
     # TODO: Handle attrs? will get x1 now
     out = x._new(data=xp.take(x._data, indices._data, axis=_axis))
     return out