Change random generation style

pandas-dev · jreback · Apr 21, 2021 · Mar 26, 2021 · Mar 26, 2021 · Mar 26, 2021
commit 482a209cacedf4ce758ae6a94d0b4dd4e88bbfdd
diff --git a/asv_bench/benchmarks/groupby.py b/asv_bench/benchmarks/groupby.py
@@ -502,18 +502,13 @@ class CumminMax:
 
     def setup(self, dtype, method):
         N = 500_000
-        rg = np.random.default_rng(seed=0)
-        vals = rg.integers(
-            0,
-            1000,
-            (N, 5),
-        )
+        vals = np.random.randint(-10, 10, (N, 5))
         null_vals = vals.astype(float, copy=True)
         null_vals[::2, :] = np.nan
         null_vals[::3, :] = np.nan
         df = DataFrame(vals, columns=list("abcde"), dtype=dtype)
         null_df = DataFrame(null_vals, columns=list("abcde"), dtype=dtype)
-        keys = rg.integers(0, 100, size=N)
+        keys = np.random.randint(0, 100, size=N)
         df["key"] = keys
         null_df["key"] = keys
         self.df = df