Merge pull request #332 from n-kall/pareto_k

fix to log weight smoothing
stan-dev · Jan 9, 2024 · 1210f4a · 1210f4a
2 parents 627d2e8 + e0c29e4
commit 1210f4a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/pareto_smooth.R b/R/pareto_smooth.R
@@ -138,7 +138,7 @@ pareto_diags.default <- function(x,
     extra_diags = TRUE,
     verbose = verbose,
     smooth_draws = FALSE,
-    are_log_weights = FALSE,
+    are_log_weights = are_log_weights,
     ...)
 
   return(smoothed$diagnostics)
@@ -337,6 +337,7 @@ pareto_smooth.default <- function(x,
       x,
       ndraws_tail = ndraws_tail,
       tail = tail,
+      are_log_weights = are_log_weights,
       ...
     )
     k <- smoothed$k