format with updated air

Author: malcolmbarrett

R/check_balance.R

@@ -193,8 +193,11 @@ check_balance <- function(
             if (any(categorical_check)) {
               categorical_cols <- original_vars_data[categorical_check]
               binary_check <- purrr::map_lgl(categorical_cols, \(x) {
-                n_levels <- if (is.factor(x)) length(levels(x)) else
+                n_levels <- if (is.factor(x)) {
+                  length(levels(x))
+                } else {
                   length(unique(x))
+                }
                 n_levels == 2
               })
               binary_categorical_names <- names(categorical_cols)[binary_check]
@@ -511,8 +514,11 @@ compute_single_balance_metric <- function(
       } else {
         setdiff(
           group_levels,
-          if (reference_group %in% group_levels) reference_group else
+          if (reference_group %in% group_levels) {
+            reference_group
+          } else {
             group_levels[reference_group]
+          }
         )[1]
       }
 

R/compute_balance.R

@@ -68,8 +68,9 @@ bal_smd <- function(
     if (is.null(weights)) {
       if (any(is.na(covariate) | is.na(group))) return(NA_real_)
     } else {
-      if (any(is.na(covariate) | is.na(group) | is.na(weights)))
+      if (any(is.na(covariate) | is.na(group) | is.na(weights))) {
         return(NA_real_)
+      }
     }
   }
 
@@ -238,8 +239,11 @@ bal_vr <- function(
       mr <- sum(wr * xr) / sum(wr)
       # Use Bessel's correction for weighted sample variance
       denom <- sum(wr) - sum(wr^2) / sum(wr)
-      if (denom <= 0) sum(wr * (xr - mr)^2) / sum(wr) else
+      if (denom <= 0) {
+        sum(wr * (xr - mr)^2) / sum(wr)
+      } else {
         sum(wr * (xr - mr)^2) / denom
+      }
     }
     var_other <- if (is.null(weights)) {
       stats::var(covariate[idx_other])
@@ -249,8 +253,11 @@ bal_vr <- function(
       mo <- sum(wo * xo) / sum(wo)
       # Use Bessel's correction for weighted sample variance
       denom <- sum(wo) - sum(wo^2) / sum(wo)
-      if (denom <= 0) sum(wo * (xo - mo)^2) / sum(wo) else
+      if (denom <= 0) {
+        sum(wo * (xo - mo)^2) / sum(wo)
+      } else {
         sum(wo * (xo - mo)^2) / denom
+      }
     }
   }
   # Return ratio
@@ -386,10 +393,16 @@ bal_ks <- function(
   # Extract and weight
   x_ref <- covariate[idx_ref]
   x_other <- covariate[idx_other]
-  w_ref <- if (is.null(weights)) rep(1, length(x_ref)) else
+  w_ref <- if (is.null(weights)) {
+    rep(1, length(x_ref))
+  } else {
     extract_weight_data(weights)[idx_ref]
-  w_other <- if (is.null(weights)) rep(1, length(x_other)) else
+  }
+  w_other <- if (is.null(weights)) {
+    rep(1, length(x_other))
+  } else {
     extract_weight_data(weights)[idx_other]
+  }
   w_ref <- w_ref / sum(w_ref)
   w_other <- w_other / sum(w_other)
   # Sort and CDF
@@ -464,7 +477,9 @@ bal_corr <- function(x, y, weights = NULL, na.rm = FALSE) {
     if (!is.null(weights)) weights <- extract_weight_data(weights)[idx]
   } else {
     # Extract weight data if needed
-    if (!is.null(weights)) weights <- extract_weight_data(weights)
+    if (!is.null(weights)) {
+      weights <- extract_weight_data(weights)
+    }
     # Check for missing values
     if (is.null(weights)) {
       if (any(is.na(x) | is.na(y))) return(NA_real_)
@@ -1053,7 +1068,9 @@ bal_energy_continuous <- function(
 
   # Avoid division by zero
   covariate_vars[covariate_vars == 0] <- 1
-  if (treatment_var == 0) treatment_var <- 1
+  if (treatment_var == 0) {
+    treatment_var <- 1
+  }
 
   # Scale covariates and treatment
   scaled_covariates <- scale(covariates, scale = sqrt(covariate_vars))

R/plot_ess.R

@@ -191,14 +191,20 @@ plot_ess <- function(
   p <- p +
     ggplot2::labs(
       x = "method",
-      y = if (percent_scale) "effective sample size (%)" else
+      y = if (percent_scale) {
+        "effective sample size (%)"
+      } else {
         "effective sample size"
+      }
     ) +
     ggplot2::scale_y_continuous(
       limits = c(0, y_upper),
       expand = c(0, 0),
-      labels = if (percent_scale) \(x) scales::percent(x / 100) else
+      labels = if (percent_scale) {
+        \(x) scales::percent(x / 100)
+      } else {
         scales::number
+      }
     )
 
   p

R/utils-validation.R

@@ -23,7 +23,9 @@ validate_weights <- function(
   arg_name = "weights",
   call = rlang::caller_env()
 ) {
-  if (is.null(weights)) return(invisible(weights))
+  if (is.null(weights)) {
+    return(invisible(weights))
+  }
 
   # Accept both numeric vectors and psw objects from propensity package
   is_valid_weights <- is.numeric(weights) || propensity::is_psw(weights)
@@ -162,15 +164,19 @@ validate_column_exists <- function(
 
 # NA handling helpers
 check_na_return <- function(..., na.rm = FALSE) {
-  if (na.rm) return(FALSE)
+  if (na.rm) {
+    return(FALSE)
+  }
 
   values <- list(...)
   any(vapply(values, function(x) any(is.na(x)), logical(1)))
 }
 
 # Filter indices based on NA values
 filter_na_indices <- function(indices, data, weights = NULL, na.rm = FALSE) {
-  if (!na.rm) return(indices)
+  if (!na.rm) {
+    return(indices)
+  }
 
   if (is.null(weights)) {
     indices[!is.na(data[indices])]