Can't get predictions from Bambi. Encountering `ValueError: different number of dimensions on data and dims: 0 vs 2`

I was running into the same error with bambi 0.16.0 but was able to fix by pinning to formulae 0.5.4.