Update groupby.py

pydata · Nov 6, 2021 · c4c4023 · c4c4023
1 parent 3df1015
commit c4c4023
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/asv_bench/benchmarks/groupby.py b/asv_bench/benchmarks/groupby.py
@@ -10,8 +10,8 @@ class GroupBy:
     def setup(self, *args, **kwargs):
         self.ds1d = xr.Dataset(
             {
-                "a": xr.DataArray(np.r_[np.arange(500.0), np.arange(500.0)]),
-                "b": xr.DataArray(np.arange(1000.0)),
+                "a": xr.DataArray(np.r_[np.arange(300), np.arange(300)]),
+                "b": xr.DataArray(np.arange(500)),
             }
         )
         self.ds2d = self.ds1d.expand_dims(z=10)
@@ -74,7 +74,7 @@ def time_agg_small_num_groups(self, method, ndim):
     @parameterized(["method", "ndim"], [("sum", "mean"), (1, 2)])
     def time_agg_large_num_groups(self, method, ndim):
         ds = getattr(self, f"ds{ndim}d")
-        getattr(ds.resample(time="6H"), method)()
+        getattr(ds.resample(time="12H"), method)()
 
 
 class ResampleDask(Resample):