Merge branch 'main' into vectorize-groupby-binary

* main:
  [skip-ci] Add benchmarks for groupby math (pydata#6390)

Author: dcherian

asv_bench/benchmarks/groupby.py

@@ -39,6 +39,12 @@ def time_groupby_binary_op_1d(self):
     def time_groupby_binary_op_2d(self):
         self.ds2d - self.ds2d_mean
 
+    def peakmem_groupby_binary_op_1d(self):
+        self.ds1d - self.ds1d_mean
+
+    def peakmem_groupby_binary_op_2d(self):
+        self.ds2d - self.ds2d_mean
+
 
 class GroupByDask(GroupBy):
     def setup(self, *args, **kwargs):
@@ -66,6 +72,9 @@ def setup(self, *args, **kwargs):
     def time_groupby_binary_op_2d(self):
         raise NotImplementedError
 
+    def peakmem_groupby_binary_op_2d(self):
+        raise NotImplementedError
+
 
 class GroupByDaskDataFrame(GroupBy):
     """Run groupby tests using dask DataFrame."""
@@ -82,6 +91,9 @@ def setup(self, *args, **kwargs):
     def time_groupby_binary_op_2d(self):
         raise NotImplementedError
 
+    def peakmem_groupby_binary_op_2d(self):
+        raise NotImplementedError
+
 
 class Resample:
     def setup(self, *args, **kwargs):
@@ -115,6 +127,12 @@ def time_groupby_binary_op_1d(self):
     def time_groupby_binary_op_2d(self):
         self.ds2d - self.ds2d_mean
 
+    def peakmem_groupby_binary_op_1d(self):
+        self.ds1d - self.ds1d_mean
+
+    def peakmem_groupby_binary_op_2d(self):
+        self.ds2d - self.ds2d_mean
+
 
 class ResampleDask(Resample):
     def setup(self, *args, **kwargs):