Add Robust SEs to a ggplot

gg_robust_ses.R

library(fabricatr)
library(randomizr)
library(estimatr)
library(ggplot2)

dat <- fabricate(N = 200,
                 Z = complete_ra(N, conditions = c("Control", "Treatment")),
                 X = sample(c("A", "B"), size = N, replace = TRUE),
                 Y = (X == "A") + (Z == "Treatment") + rnorm(N, sd = 1 + (Z == "Treatment")))

ggplot(dat, aes(Z, Y, group = X, color = X)) +
  geom_point(position = position_jitter(height = 0, width = 0.1), alpha = .1) +
  stat_smooth(method = "lm_robust") +
  theme_bw() +
  ggtitle("Robust SEs on a ggplot is easy with estimatr::lm_robust()",
          'Just do stat_smooth(method = "lm_robust")')