From dc68595a4f3ec3848661cb709064b0a7000af7dd Mon Sep 17 00:00:00 2001 From: Drew Herren Date: Thu, 20 Jun 2024 17:53:56 -0500 Subject: [PATCH] Fixed small BCF bug that shows up in serialization roundtrip --- R/bcf.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/bcf.R b/R/bcf.R index 8275a734..65cef96c 100644 --- a/R/bcf.R +++ b/R/bcf.R @@ -768,7 +768,7 @@ bcf <- function(X_train, Z_train, y_train, pi_train = NULL, group_ids_train = NU "b_leaf_tau" = b_leaf_tau, "outcome_mean" = y_bar_train, "outcome_scale" = y_std_train, - "num_covariates" = ncol(X_train), + "num_covariates" = num_cov_orig, "num_prognostic_covariates" = sum(variable_weights_mu > 0), "num_treatment_covariates" = sum(variable_weights_tau > 0), "treatment_dim" = ncol(Z_train),