From bef6ac0482b2270252f64be5e6212c331a8d30c3 Mon Sep 17 00:00:00 2001 From: Osvaldo Martin Date: Wed, 17 Apr 2019 14:33:47 -0300 Subject: [PATCH] make waic return a Series (#646) --- arviz/stats/stats.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/arviz/stats/stats.py b/arviz/stats/stats.py index cb23650c4b..f4b318e8f3 100644 --- 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"], )