What I would really like is to type check the shape of a numpy array to prevent dimension array at runtime. You could use third party library like https://github.com/ramonhagenaars/nptyping or https://github.com/beartype/beartype#numpy-arrays but it will not extend to the methode of Numpy.
Maybe you are looking for something like this? https://news.ycombinator.com/item?id=40022628
But not sure how far this gets you with regards to the pain points mentioned in this post.