small cleanups

asv_bench/benchmarks/reduce.py

@@ -10,7 +10,6 @@
 engines = ["flox", "numpy", "numbagg"]
 expected_groups = {
     "None": None,
-    # "RangeIndex": pd.RangeIndex(5),
     "bins": pd.IntervalIndex.from_breaks([1, 2, 4]),
 }
 expected_names = tuple(expected_groups)
@@ -95,14 +94,6 @@ def time_reduce_bare(self, func, engine):
         )
 
 
-# class ChunkReduce1DUnsorted(ChunkReduce):
-#     def setup(self, *args, **kwargs):
-#         self.array = np.ones((N,))
-#         self.labels = np.random.permutation(np.repeat(np.arange(5), repeats=N // 5))
-#         self.axis = -1
-#         setup_jit()
-
-
 class ChunkReduce2D(ChunkReduce):
     def setup(self, *args, **kwargs):
         self.array = np.ones((N, N))
@@ -111,20 +102,27 @@ def setup(self, *args, **kwargs):
         setup_jit()
 
 
+class ChunkReduce2DAllAxes(ChunkReduce):
+    def setup(self, *args, **kwargs):
+        self.array = np.ones((N, N))
+        self.labels = np.repeat(np.arange(N // 5), repeats=5)
+        self.axis = None
+        setup_jit()
+
+
 # class ChunkReduce2DUnsorted(ChunkReduce):
 #     def setup(self, *args, **kwargs):
 #         self.array = np.ones((N, N))
 #         self.labels = np.random.permutation(np.repeat(np.arange(N // 5), repeats=5))
 #         self.axis = -1
 #         setup_jit()
 
-
-class ChunkReduce2DAllAxes(ChunkReduce):
-    def setup(self, *args, **kwargs):
-        self.array = np.ones((N, N))
-        self.labels = np.repeat(np.arange(N // 5), repeats=5)
-        self.axis = None
-        setup_jit()
+# class ChunkReduce1DUnsorted(ChunkReduce):
+#     def setup(self, *args, **kwargs):
+#         self.array = np.ones((N,))
+#         self.labels = np.random.permutation(np.repeat(np.arange(5), repeats=N // 5))
+#         self.axis = -1
+#         setup_jit()
 
 
 # class ChunkReduce2DAllAxesUnsorted(ChunkReduce):