workaround a shape mismatch

also reformatted the call to interpolation

---

This np.repeat seems to be causing a fault when running the script, any insights? /cc @xuchuang