Ritvik, have had a chance to update this using PyMC v5? and also the new versions of StatsModels.tsa.arima? There are some massive differences and your code no longer works. Thank you.
btw, the original code worked perfectly. Thank you for your contributions.