plot_out_of_sample_model_fit shape broadcast issue

[mike-duran-mitchell]

trafficstars

Running into an error where I am following the tutorial at the example at https://github.com/google/lightweight_mmm/blob/main/examples/simple_end_to_end_demo.ipynb but when I use

plot.plot_out_of_sample_model_fit(out_of_sample_predictions=new_predictions,
                                 out_of_sample_target=target_scaler.transform(target[split_point:]))

I am getting an error ValueError: Length of values (2000) does not match length of index (13)

I've tried reshaping the data so that it uses just the mean and then predictions and target are the same shape of (13,), but then it gives me an error about IndexError: tuple index out of range. Any ideas?