Use output element nprjdraws consistently.

stan-dev · Dec 7, 2023 · 1bce71e · 1bce71e
1 parent a15bbe4
commit 1bce71e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/varsel.R b/R/varsel.R
@@ -483,7 +483,7 @@ varsel.refmodel <- function(object, d_test = NULL, method = "forward",
               ),
               clust_used_search = search_path$p_sel$clust_used,
               clust_used_eval = perf_eval_out[["clust_used"]],
-              nprjdraws_search = NCOL(search_path$p_sel$mu),
+              nprjdraws_search = search_path$p_sel$nprjdraws,
               nprjdraws_eval = perf_eval_out[["nprjdraws"]],
               refit_prj,
               projpred_version = utils::packageVersion("projpred"))