nomatch force for the no macthing leaf case

bips-hb · Jun 13, 2024 · 8d294f2 · 8d294f2
1 parent 6b28747
commit 8d294f2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/R/utils.R b/R/utils.R
@@ -436,7 +436,15 @@ cforde <- function(params,
         }
       }
       conds_impossible <- conds_conditioned[!(conds_conditioned %in% relevant_leaves[,unique(c_idx)])]
-      relevant_leaves <- setorder(rbind(relevant_leaves, data.table(c_idx = conds_impossible, f_idx = NA_integer_, f_idx_uncond = NA_integer_)))
+
+      if (grepl("^force", nomatch)) {
+        # All leaves 
+        impossible_leaves <- data.table(c_idx = conds_impossible, f_idx = forest$f_idx, f_idx_uncond = forest$f_idx)
+      } else {
+        # Set to NA -> no leaves -> Sample NA
+        impossible_leaves <- data.table(c_idx = conds_impossible, f_idx = NA_integer_, f_idx_uncond = NA_integer_)
+      }
+      relevant_leaves <- setorder(rbind(relevant_leaves, impossible_leaves))
     }
   }