add skip_trailing_dim to _ravel_shape_indices

dask · Jul 12, 2024 · ea976ab · ea976ab
1 parent 09aabef
commit ea976ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dask_image/ndmeasure/__init__.py b/dask_image/ndmeasure/__init__.py
@@ -442,7 +442,7 @@ def labeled_comprehension(image,
     args = (image,)
     if pass_positions:
         positions = _utils._ravel_shape_indices(
-            image.shape, chunks=image.chunks
+            image.shape, chunks=image.chunks, skip_trailing_dim=image.ndim != label_image.ndim
         )
         args = (image, positions)