Update `check_ndarray_or_raise` in argcheck.py