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.