make waic return a Series (#646)

arviz-devs · Apr 17, 2019 · bef6ac0 · bef6ac0
1 parent 1c1a236
commit bef6ac0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/arviz/stats/stats.py b/arviz/stats/stats.py
@@ -1030,12 +1030,12 @@ def waic(data, pointwise=False, scale="deviance"):
             the Observed RV in your model to make sure it returns element-wise logp.
             """
             )
-        return pd.DataFrame(
-            [[waic_sum, waic_se, p_waic, warn_mg, waic_i, scale]],
-            columns=["waic", "waic_se", "p_waic", "warning", "waic_i", "waic_scale"],
+        return pd.Series(
+            data=[waic_sum, waic_se, p_waic, warn_mg, waic_i, scale],
+            index=["waic", "waic_se", "p_waic", "warning", "waic_i", "waic_scale"],
         )
     else:
-        return pd.DataFrame(
-            [[waic_sum, waic_se, p_waic, warn_mg, scale]],
-            columns=["waic", "waic_se", "p_waic", "warning", "waic_scale"],
+        return pd.Series(
+            data=[waic_sum, waic_se, p_waic, warn_mg, scale],
+            index=["waic", "waic_se", "p_waic", "warning", "waic_scale"],
         )