From ef4d236d7a3b6529010796271fac31d75e5cccac Mon Sep 17 00:00:00 2001
From: Aki Vehtari <Aki.Vehtari@aalto.fi>
Date: Sat, 17 Feb 2024 19:22:33 +0200
Subject: [PATCH] show NA for n_eff/ESS if k > k_threshold

---
 R/diagnostics.R | 1 +
 1 file changed, 1 insertion(+)

diff --git a/R/diagnostics.R b/R/diagnostics.R
index 33dd66f1..a7b489ac 100644
--- a/R/diagnostics.R
+++ b/R/diagnostics.R
@@ -130,6 +130,7 @@ pareto_k_table <- function(x) {
   S <- dim(x)[1]
   k_threshold <- ps_khat_threshold(S)
   kcut <- k_cut(k, k_threshold)
+  n_eff[k>k_threshold] <- NA
   min_n_eff <- min_n_eff_by_k(n_eff, kcut)
   count <- table(kcut)
   out <- cbind(