Revert "Refactor run function to aggregate batches and update command-line arguments"

This reverts commit 7b867fe.

Author: kosiew

benchmarks/collect_gil_bench.py

@@ -21,8 +21,7 @@
 import time
 
 import pyarrow as pa
-from datafusion import SessionContext, col
-from datafusion import functions as f
+from datafusion import SessionContext
 
 
 def run(
@@ -47,36 +46,20 @@ def run(
     df = ctx.create_dataframe(partitions)
 
     start = time.perf_counter()
-    df.aggregate([], [f.sum(col("a"))]).collect()
+    df.collect()
     duration = time.perf_counter() - start
-    print(f"{n_batches} batches aggregated in {duration:.3f}s")
+    print(f"{n_batches} batches collected in {duration:.3f}s")
 
 
 if __name__ == "__main__":
     import argparse
 
     parser = argparse.ArgumentParser()
-    parser.add_argument(
-        "--batches",
-        type=int,
-        default=8,
-        help="number of input batches to generate",
-    )
-    parser.add_argument(
-        "--batch-size",
-        type=int,
-        default=1_000_000,
-        help="number of rows per batch",
-    )
     parser.add_argument(
         "--partitions",
         type=int,
         default=None,
         help="number of partitions to create (defaults to one per batch)",
     )
     args = parser.parse_args()
-    run(
-        n_batches=args.batches,
-        batch_size=args.batch_size,
-        n_partitions=args.partitions,
-    )
+    run(n_partitions=args.partitions)