Refactor batch handling in collect_gil_bench.py to use partitions for better organization

Author: kosiew

benchmarks/collect_gil_bench.py

@@ -23,13 +23,14 @@
 
 def run(n_batches: int = 8, batch_size: int = 1_000_000) -> None:
     ctx = SessionContext()
-    batches = []
+    partitions = []
     for i in range(n_batches):
         start = i * batch_size
         arr = pa.array(range(start, start + batch_size))
-        batches.append(pa.record_batch([arr], names=["a"]))
+        batch = pa.record_batch([arr], names=["a"])
+        partitions.append([batch])  # Each batch in its own partition
 
-    df = ctx.create_dataframe([batches])
+    df = ctx.create_dataframe(partitions)
 
     start = time.perf_counter()
     df.collect()