Remove duplicate axis check

Co-authored-by: Eric Wieser <[email protected]>
kulinseth · Jan 23, 2021 · 5cbfefb · 5cbfefb
1 parent 768acb7
commit 5cbfefb
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/numpy/random/_generator.pyx b/numpy/random/_generator.pyx
@@ -4416,10 +4416,6 @@ cdef class Generator:
                     _shuffle_raw_wrap(&self._bitgen, n, 1, itemsize, stride,
                                       x_ptr, buf_ptr)
         elif isinstance(x, np.ndarray):
-            if axis >= x.ndim:
-                raise np.AxisError(f"Cannot shuffle along axis {axis} for "
-                                   f"array of dimension {x.ndim}")
-
             if x.size == 0:
                 # shuffling is a no-op
                 return