fix issue with plotting z-scored images

ANTsX · Mar 12, 2024 · fe3783b · fe3783b
1 parent b8e7118
commit fe3783b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ants/viz/plot.py b/ants/viz/plot.py
@@ -228,7 +228,7 @@ def reorient_slice(x, axis):
     if not isinstance(image, iio.ANTsImage):
         raise ValueError("image argument must be an ANTsImage")
 
-    if not image.sum() > 0:
+    if sum(sum(image != 0)) == 0:
         warnings.warn("Image must be non-zero. will not plot.")
         return