Update app.R

app.R

@@ -136,7 +136,8 @@ server <- function(input, output, session) {
                              my_data$Office == "President")
         }
         
-        FACTOR_MAT <- FACTOR_MAT_FULL[indices, !colnames(FACTOR_MAT_FULL) %in% "Office"]
+        FACTOR_MAT <- FACTOR_MAT_FULL[indices, 
+                      !colnames(FACTOR_MAT_FULL) %in% c("Office","Party.affiliation","Party.competition")]
         Yobs <- Yobs_FULL[indices]
         X <- X_FULL[indices, ]
         log_pr_w <- log_pr_w_FULL[indices]
@@ -152,7 +153,7 @@ server <- function(input, output, session) {
           X = X,
           pair_id = pair_id,
           
-          p_list = assignmentProbList,
+          p_list = assignmentProbList[colnames(FACTOR_MAT)],
           lambda = my_lambda,
           diff = TRUE, 
           adversarial = FALSE,