Update tests and baselines for sumstats docs