NA rows in df will cause uninformative error for brsmatch()

[skent259]

Simple test case. Problem is that the data frame output in .compute_distances() will have different number of rows since model.matrix() will automatically remove NAs.

library(rsmatch)

df <- data.frame(
  id = rep(1:3, each = 3),
  time = rep(1:3, 3),
  trt_time = rep(c(2, 3, NA), each = 3),
  X1 = c(2, 2, 2, 3, 3, 3, 9, 9, 9),
  X2 = rep(c("a", "a", "b"), each = 3),
  X3 = c(9, 4, 5, 6, NA, NA, 3, 4, 8),
  X4 = c(8, 9, 4, 5, 6, 7, 2, 3, 4)
)

brsmatch(n_pairs = 1, data = df)
#> Error in data.frame(trt_id = i, all_id = df_at_trt[[id]], trt_time = trt_time_i, : arguments imply differing number of rows: 1, 3, 2

^{Created on 2024-02-03 with reprex v2.0.2}

[skent259]

Confirmed that similar problem will happen with coxpsmatch() function