isTRUE

andrjohns · andrjohns · commit 2ba45955acca · 2025-10-04T22:10:21.000+08:00
diff --git a/R/sample.R b/R/sample.R
@@ -263,7 +263,7 @@ stan_sample <- function(fn, par_inits = NULL, n_pars = NULL, additional_args = l
   par_vars <- draw_names[!(draw_names %in% diagnostic_vars)]
   draws <- posterior::as_draws_df(do.call(rbind.data.frame, draws))
   diagnostics <- posterior::subset_draws(draws, variable = diagnostic_vars)
-  if (save_warmup) {
+  if (isTRUE(save_warmup)) {
     diagnostics <- diagnostics[diagnostics$.iteration > num_warmup, ]
   }
   if (check_diagnostics) {

Original file line number	Diff line number	Diff line change
`@@ -263,7 +263,7 @@ stan_sample <- function(fn, par_inits = NULL, n_pars = NULL, additional_args = l`
`263`	`263`	`par_vars <- draw_names[!(draw_names %in% diagnostic_vars)]`
`264`	`264`	`draws <- posterior::as_draws_df(do.call(rbind.data.frame, draws))`
`265`	`265`	`diagnostics <- posterior::subset_draws(draws, variable = diagnostic_vars)`
`266`		`- if (save_warmup) {`
	`266`	`+ if (isTRUE(save_warmup)) {`
`267`	`267`	`diagnostics <- diagnostics[diagnostics$.iteration > num_warmup, ]`
`268`	`268`	`}`
`269`	`269`	`if (check_diagnostics) {`