Forecasting with python

My project goal is to produce a forecast using Python on a given dataset of teams. It is a short timeseries. The train dataset is 78 months; the test dataset is 12 months. I used jupyter to write code that employs sarima, prophet and xgb. I would like someone to look at the code, write it more efficiently and if possible improve the forecasting accuracy for all teams.