epidemic-forecasting-for-python