Update python/cucim/src/cucim/skimage/_vendored/_ndimage_util.py

Co-authored-by: Gigon Bae <[email protected]>
rapidsai · Jan 21, 2025 · c366868 · c366868
1 parent af2fce4
commit c366868
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cucim/src/cucim/skimage/_vendored/_ndimage_util.py b/python/cucim/src/cucim/skimage/_vendored/_ndimage_util.py
@@ -109,8 +109,8 @@ def _check_axes(axes, ndim):
     elif cupy.isscalar(axes):
         axes = (operator.index(axes),)
     elif isinstance(axes, Iterable):
+        axes = tuple(operator.index(ax) for ax in axes)
         for ax in axes:
-            axes = tuple(operator.index(ax) for ax in axes)
             if ax < -ndim or ax > ndim - 1:
                 raise AxisError(f"specified axis: {ax} is out of range")
         axes = tuple(ax % ndim if ax < 0 else ax for ax in axes)