Skip to content

Commit 4a8cdb2

Browse files
Update column names in pdsh query 18 (#21119)
This updates the column names in query 18 to match other engines. Validated locally with ``` ❯ python -m cudf_polars.experimental.benchmarks.pdsh --path /datasets/toaugspurger/tpch-rs/scale-10/ --suffix "" --iterations 1 18 --validate ❯ python -m cudf_polars.experimental.benchmarks.pdsh --path /datasets/toaugspurger/tpch-rs/scale-10/ --suffix "" --iterations 1 18 --engine duckdb ``` Authors: - Tom Augspurger (https://github.com/TomAugspurger) - Benjamin Zaitlen (https://github.com/quasiben) - Matthew Murray (https://github.com/Matt711) Approvers: - Matthew Murray (https://github.com/Matt711) - Richard (Rick) Zamora (https://github.com/rjzamora) URL: #21119
1 parent 8c2a45e commit 4a8cdb2

File tree

1 file changed

+7
-7
lines changed
  • python/cudf_polars/cudf_polars/experimental/benchmarks

1 file changed

+7
-7
lines changed

python/cudf_polars/cudf_polars/experimental/benchmarks/pdsh.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES.
1+
# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES.
22
# SPDX-License-Identifier: Apache-2.0
33

44
"""
@@ -648,16 +648,16 @@ def q18(run_config: RunConfig) -> pl.LazyFrame:
648648
.group_by(
649649
"c_name", "o_custkey", "o_orderkey", "o_orderdate", "o_totalprice"
650650
)
651-
.agg(pl.col("l_quantity").sum().alias("col6"))
651+
.agg(pl.col("l_quantity").sum().alias("sum(l_quantity)"))
652652
.select(
653653
pl.col("c_name"),
654654
pl.col("o_custkey").alias("c_custkey"),
655655
pl.col("o_orderkey"),
656-
pl.col("o_orderdate").alias("o_orderdat"),
656+
pl.col("o_orderdate"),
657657
pl.col("o_totalprice"),
658-
pl.col("col6"),
658+
pl.col("sum(l_quantity)"),
659659
)
660-
.sort(by=["o_totalprice", "o_orderdat"], descending=[True, False])
660+
.sort(by=["o_totalprice", "o_orderdate"], descending=[True, False])
661661
.head(100)
662662
)
663663

@@ -1373,9 +1373,9 @@ def q18(run_config: RunConfig) -> str:
13731373
c_name,
13741374
c_custkey,
13751375
o_orderkey,
1376-
o_orderdate as o_orderdat,
1376+
o_orderdate,
13771377
o_totalprice,
1378-
sum(l_quantity) as col6
1378+
sum(l_quantity)
13791379
from
13801380
customer,
13811381
orders,

0 commit comments

Comments
 (0)