Might be worth pointing out in a comment and/or the error message this is a numpy limitation.