plots: allow 3d and 4d when plotting multiples explanations

deel-ai · Nov 2, 2022 · 5c60d51 · 5c60d51
1 parent fdea9f8
commit 5c60d51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xplique/plots/image.py b/xplique/plots/image.py
@@ -153,7 +153,7 @@ def plot_attributions(
 
     rows = ceil(len(explanations) / cols)
     # get width and height of our images
-    l_width, l_height = explanations.shape[1:]
+    l_width, l_height = explanations.shape[1:3]
 
     # define the figure margin, width, height in inch
     margin = 0.3