Update _set_functions.py

pydata · Sep 24, 2024 · 6cf9fae · 6cf9fae
1 parent 00115d1
commit 6cf9fae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xarray/namedarray/_array_api/_set_functions.py b/xarray/namedarray/_array_api/_set_functions.py
@@ -34,7 +34,7 @@ def unique_all(x: NamedArray[Any, Any], /) -> UniqueAllResult:
     return UniqueAllResult(
         NamedArray(_dims, values),
         NamedArray(_dims, indices),
-        NamedArray(_dims, inverse_indices),
+        NamedArray(_flatten_dims(x.dims), inverse_indices),
         NamedArray(_dims, counts),
     )