fix shape bug in convex_hull when 2d image has a singletone size on o…

…ne axis
rapidsai · Mar 1, 2025 · 783a434 · 783a434
1 parent d7e724c
commit 783a434
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cucim/src/cucim/skimage/morphology/convex_hull.py b/python/cucim/src/cucim/skimage/morphology/convex_hull.py
@@ -207,7 +207,7 @@ def convex_hull_image(
     original_shape = image.shape
     image = cp.squeeze(image)
     if image.ndim < 2:
-        return image
+        return image.reshape(original_shape)
 
     if image.size < cpu_fallback_threshold:
         # Fallback to pure CPU implementation