Handle NAs in orientation

Author: cjerzak

lpme/R/lpme_DoBootPartition.R

@@ -160,10 +160,10 @@ lpme <- function(Y,
     if (!all(orientation_signs %in% c(1, -1))) {
       stop("orientation_signs must contain only 1 and -1.")
     }
-    if(!all(unlist(observables) %in% c(0,1))){
+    if(!all(na.omit(unlist(observables)) %in% c(0,1))){
       stop("Re-orientation in the non-binary case not yet implemented")
     }
-    if(all(observables %in% c(0,1))){
+    if(all(na.omit(unlist(observables)) %in% c(0,1))){
       colnames_observables <- colnames(observables)
       observables <- sapply(1:ncol(observables), function(d_){
         observables[, d_] <- orientation_signs[d_] * observables[, d_] + 

lpme/R/orientation_helpers.R

@@ -28,7 +28,9 @@
 #' @export
 infer_orientation_signs <- function(Y, observables, method = c("Y", "PC1")) {
   method <- match.arg(method)
-  if (!all(unlist(observables) %in% c(0, 1))) {
+  # allow for missing values when validating the observables
+  obs_vals <- na.omit(unlist(observables))
+  if (!all(obs_vals %in% c(0, 1))) {
     stop("infer_orientation_signs currently supports only binary observables")
   }
   if (method == "PC1") {