ci: improve db-benchmark q8 (~2x faster) (#4614)

pola-rs · Aug 29, 2022 · 164936a · 164936a
1 parent 65e2f85
commit 164936a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/py-polars/tests/db-benchmark/main.py b/py-polars/tests/db-benchmark/main.py
@@ -119,9 +119,8 @@
 print("q8")
 out = (
     x.drop_nulls("v3")
-    .sort("v3", reverse=True)
     .groupby("id6")
-    .agg(pl.col("v3").head(2).alias("largest2_v3"))
+    .agg(pl.col("v3").top_k(2).alias("largest2_v3"))
     .explode("largest2_v3")
     .collect()
 )