Mixup causes bug in Interpretation functions?

I found something peculiar.

Running the lesson1-pets.ipynb for example, I can visualize the heatmap without issues.

When I add mixup “learn = learn.mixup”, for some reason commands like learn.get_preds return a rank 0 tensor instead of a rank 1 tensor. This causes all downstream interpretation function calls to fail.

Is anyone seeing this?