fix moments labels (#471)

arviz-devs · Jun 17, 2024 · eac0928 · eac0928
1 parent ca01bf6
commit eac0928
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/preliz/internal/plot_helper.py b/preliz/internal/plot_helper.py
@@ -266,11 +266,10 @@ def get_moments(dist, moments):
     }
     str_m = []
     seen = []
-    for moment in moments:
+    values = dist.moments(moments)
+    for moment, value in zip(moments, values):
         if moment not in seen:
-            value = dist.moments(moment)
-            if isinstance(value, (np.ndarray, int, float)):
-                str_m.append(f"{names[moment]}={value:.3g}")
+            str_m.append(f"{names[moment]}={value:.3g}")
 
         seen.append(moment)