Add context support for forecast and deploy