Revert "Temporary comment ARIMA train and forecast"

This reverts commit 7c4f388efe9b36ca7594c4ec47d20d88945cd207.