Could we use numpy or python warning instead?