mcmc_pairs: >= max_treedepth instead of > max_treedepth

fixes #281
stan-dev · Nov 14, 2022 · 6669943 · 6669943
1 parent b38d468
commit 6669943
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/mcmc-scatterplots.R b/R/mcmc-scatterplots.R
@@ -351,7 +351,7 @@ mcmc_pairs <- function(x,
     divs <- dplyr::filter(np, UQ(param) == "divergent__") %>% pull(UQ(val))
     divergent__ <- matrix(divs, nrow = n_iter * n_chain, ncol = n_param)[, 1]
     if (!no_max_td) {
-      gt_max_td <- (dplyr::filter(np, UQ(param) == "treedepth__") %>% pull(UQ(val))) > max_treedepth
+      gt_max_td <- (dplyr::filter(np, UQ(param) == "treedepth__") %>% pull(UQ(val))) >= max_treedepth
       max_td_hit__ <- matrix(gt_max_td, nrow = n_iter * n_chain, ncol = n_param)[, 1]
     }
   }