Great, seems like a simple fix. It’s just the mismatched array shapes that creates an n x n array instead of doing element-wise division.
I can put together a PR with that one-line change. It would be better to have it in the main repo than worrying about tracking a local version.