cleanup

pydata · Dec 3, 2024 · 6f47c5b · 6f47c5b
1 parent 1271e88
commit 6f47c5b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/xarray/core/indexing.py b/xarray/core/indexing.py
@@ -1648,9 +1648,15 @@ def _vindex_get(self, indexer: VectorizedIndexer):
         except IndexError as e:
             # TODO: upstream to dask
             has_dask = any(is_duck_dask_array(i) for i in indexer.tuple)
-            if not has_dask or (has_dask and len(indexer.tuple) > 1):
-                raise e
-            if math.prod(self.array.numblocks) > 1 or self.array.ndim > 1:
+            # this only works for "small" 1d coordinate arrays with one chunk
+            # it is intended for idxmin, idxmax, and allows indexing with
+            # the output of argmin, argmax
+            if (
+                not has_dask
+                or len(indexer.tuple) > 1
+                or math.prod(self.array.numblocks) > 1
+                or self.array.ndim > 1
+            ):
                 raise e
             (idxr,) = indexer.tuple
             if idxr.ndim == 0: