Uncertainty quantification for data-driven weather models