Impute results in mismatch dimensions in dims and data

hmm. the result here though is that I’ve lost the original dimensionality for load. Meaning that to add this to the original source dataset, I need to do quite a lot of transformation on the inference_data object. Defeating the original purpose of using the wide data :slight_smile:

The goal was to be able to use something like az.plot_lm() and target a specific exercise and date combination for the plot.