fix #161: avoid Inf nclusters_pred when both NULL

stan-dev · Jun 27, 2021 · 38d7621 · 38d7621
1 parent 76d22a1
commit 38d7621
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/R/cv_varsel.R b/R/cv_varsel.R
@@ -549,10 +549,12 @@ kfold_varsel <- function(refmodel, method, nterms_max, ndraws,
   ## List of K elements, each containing d_train, p_pred, etc. corresponding
   ## to each fold.
   make_list_cv <- function(refmodel, d_test, msg) {
-    nclusters_pred <- min(
-      refmodel$nclusters_pred,
-      nclusters_pred
-    )
+    if (!is.null(nclusters_pred) || !is.null(refmodel$nclusters_pred)) {
+      nclusters_pred <- min(
+        refmodel$nclusters_pred,
+        nclusters_pred
+      )
+    }
     p_sel <- .get_refdist(refmodel, ndraws, nclusters, seed = seed)
     p_pred <- .get_refdist(refmodel, ndraws_pred, nclusters_pred, seed = seed)
     newdata <- d_test$newdata