rotate: fix comparison bug

dask · Nov 8, 2023 · a371ebc · a371ebc
1 parent a9c7a15
commit a371ebc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dask_image/ndinterp/__init__.py b/dask_image/ndinterp/__init__.py
@@ -407,7 +407,7 @@ def rotate(input_arr, angle, axes=(1, 0), reshape=True, output=None, order=1,
     else:
         out_plane_shape = img_shape[axes]
 
-    if output_shape == img_shape:
+    if (output_shape == img_shape).all():
         output_shape[axes] = out_plane_shape
     else:
         out_plane_shape = np.asarray(output_shape)[axes]