diff --git a/R/bcf.R b/R/bcf.R index 921b7570..4a645986 100644 --- a/R/bcf.R +++ b/R/bcf.R @@ -596,6 +596,7 @@ bcf <- function(X_train, Z_train, y_train, pi_train = NULL, group_ids_train = NU } # Update feature_types and covariates + feature_types <- as.integer(feature_types) if (propensity_covariate != "none") { feature_types <- as.integer(c(feature_types,0)) X_train <- cbind(X_train, pi_train)