X (DataArray) – Baseline design matrix (n_obs, n_baseline_features).
y (DataArray) – Outcome variable (n_obs, 1).
coords (dict[str, Any]) – Coordinate names for PyMC.
treatment_data (DataArray) – Raw treatment variables (n_obs, n_treatments).