sample() does not work for vector with length 1

bips-hb · Jan 17, 2024 · fa3dd7e · fa3dd7e
1 parent c2d7f27
commit fa3dd7e
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/R/forge.R b/R/forge.R
@@ -108,12 +108,14 @@ forge <- function(
 
   if (nrow(omega) == 0) {
     stop("All leaves have zero likelihood.")
+  } else if (nrow(omega) == 1) {
+    draws <- omega[, .(f_idx)]
+  } else {
+    # Draw random leaves with probability proportional to weight
+    draws <- data.table(
+      'f_idx' = omega[, sample(f_idx, size = n_synth, replace = TRUE, prob = wt)]
+    )
   }
-
-  # Draw random leaves with probability proportional to weight
-  draws <- data.table(
-    'f_idx' = omega[, sample(f_idx, size = n_synth, replace = TRUE, prob = wt)]
-  )
   omega <- merge(draws, omega, sort = FALSE)[, idx := .I]
 
   # Simulate continuous data