From 6669943bcfc1c4740d0632821341b5a29d4d3371 Mon Sep 17 00:00:00 2001 From: jgabry Date: Mon, 14 Nov 2022 16:32:17 -0700 Subject: [PATCH] mcmc_pairs: >= max_treedepth instead of > max_treedepth fixes #281 --- R/mcmc-scatterplots.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/mcmc-scatterplots.R b/R/mcmc-scatterplots.R index 358cdc72..7bb80ab2 100644 --- 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] } }