Ran under AZ databrick.
here is the code: 
#Select SARIMA with seasonality 12
autoARIMA = AutoARIMA(season_length=12)

# Select ETS with seasonality 12 and multiplicative trend
model = StatsForecast(df=products2.set_index('unique_id'), 
                      models=[autoARIMA],
                      freq='m', n_jobs=-1)
Y_hat_df = model.forecast(horizon).reset_index()