Update calc_bal_stats.R

istallworthy · istallworthy · commit aa5534a0037c · 2026-04-13T08:21:01.000-04:00
omitting bal_stats_histories that are all 1s for balance stats weighting
diff --git a/R/calc_bal_stats.R b/R/calc_bal_stats.R
@@ -123,11 +123,12 @@ calc_bal_stats <- function(data, obj, weights = NULL, balance_thresh = NULL, imp
         )
       } # ends hist exc
       
-      # added by IS 4/8/26 to skip histories for which exposures are all 0s (cannot calc bal stats)
+      # added by IS 4/8/26 to skip histories for which exposures are all 0s or all 1s (cannot calc bal stats)
       omitted_histories <- Filter(
         function(h) {
           which_idx <- (history == h)
-          sum(data[[exposure_name]][which_idx]) == 0
+          vals <- data[[exposure_name]][which_idx]
+          all(vals == 0) || all(vals == 1)
         },
         setdiff(prop_sum$history, omitted_histories)
       )

Original file line number	Diff line number	Diff line change
`@@ -123,11 +123,12 @@ calc_bal_stats <- function(data, obj, weights = NULL, balance_thresh = NULL, imp`
`123`	`123`	`)`
`124`	`124`	`} # ends hist exc`
`125`	`125`
`126`		`- # added by IS 4/8/26 to skip histories for which exposures are all 0s (cannot calc bal stats)`
	`126`	`+ # added by IS 4/8/26 to skip histories for which exposures are all 0s or all 1s (cannot calc bal stats)`
`127`	`127`	`omitted_histories <- Filter(`
`128`	`128`	`function(h) {`
`129`	`129`	`which_idx <- (history == h)`
`130`		`- sum(data[[exposure_name]][which_idx]) == 0`
	`130`	`+ vals <- data[[exposure_name]][which_idx]`
	`131`	`+ all(vals == 0) \|\| all(vals == 1)`
`131`	`132`	`},`
`132`	`133`	`setdiff(prop_sum$history, omitted_histories)`
`133`	`134`	`)`