Issue with numba: “cannot unify array” unless I convert scalars to array

Question

What I am trying to do

I am trying to create a very simple function which I want to optimise with numba (or at least verify if numba makes any difference).

I am