fix unconditional forge

bips-hb · Feb 9, 2024 · 7cead3a · 7cead3a
1 parent c09b12f
commit 7cead3a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/R/forge.R b/R/forge.R
@@ -99,6 +99,7 @@ forge <- function(
     omega <- params$forest[, .(f_idx, cvg)]
     omega[, wt := cvg / num_trees]
     omega[, cvg := NULL]
+    omega[, row_idx := 1]
   } else if (isTRUE(conj)) {
     omega <- leaf_posterior(params, evidence)
   } else {