diff options
Diffstat (limited to 'numpy/random/mtrand.pyx')
| -rw-r--r-- | numpy/random/mtrand.pyx | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/random/mtrand.pyx b/numpy/random/mtrand.pyx index 9eabfd848..6d36dda36 100644 --- a/numpy/random/mtrand.pyx +++ b/numpy/random/mtrand.pyx @@ -4198,11 +4198,11 @@ cdef class RandomState: cdef long *mnix cdef long ni - if np.ndim(pvals) != 1: - raise ValueError("pvals must be 1d array") d = len(pvals) parr = <np.ndarray>np.PyArray_FROM_OTF( pvals, np.NPY_DOUBLE, np.NPY_ALIGNED | np.NPY_ARRAY_C_CONTIGUOUS) + if np.PyArray_NDIM(parr) != 1: + raise ValueError("pvals must be 1d array") pix = <double*>np.PyArray_DATA(parr) check_array_constraint(parr, 'pvals', CONS_BOUNDED_0_1) if kahan_sum(pix, d-1) > (1.0 + 1e-12): |