Merge pull request #9 from ck37/nnloglik_mod_fix

nnloglik_mod: ensure modZ remains a matrix after dropping duplicate columns
benkeser · May 14, 2018 · f4cbd24 · f4cbd24
2 parents a14d72f + 0d67e39
commit f4cbd24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/method_fixes.R b/R/method_fixes.R
@@ -102,7 +102,7 @@ method.CC_nloglik_mod <- function() {
         "Algorithm ", colDup,
         " is duplicated. Setting weight to 0."
       ))
-      modZ <- modZ[, -colDup]
+      modZ <- modZ[, -colDup, drop = FALSE]
     }
     modlogitZ <- SuperLearner::trimLogit(modZ, control$trimLogit)
     logitZ <- SuperLearner::trimLogit(Z, control$trimLogit)