Skip to content

Commit 2659ea6

Browse files
chore: Amend tpch q9 to more closely resemble sql (#1415)
1 parent 693ed8c commit 2659ea6

File tree

1 file changed

+9
-5
lines changed
  • third_party/bigframes_vendored/tpch/queries

1 file changed

+9
-5
lines changed

third_party/bigframes_vendored/tpch/queries/q9.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,17 @@ def q(project_id: str, dataset_id: str, session: bigframes.Session):
3333
)
3434

3535
q_final = (
36-
part.merge(partsupp, left_on="P_PARTKEY", right_on="PS_PARTKEY")
37-
.merge(supplier, left_on="PS_SUPPKEY", right_on="S_SUPPKEY")
38-
.merge(
36+
part.merge(
3937
lineitem,
40-
left_on=["P_PARTKEY", "PS_SUPPKEY"],
41-
right_on=["L_PARTKEY", "L_SUPPKEY"],
38+
left_on="P_PARTKEY",
39+
right_on="L_PARTKEY",
40+
)
41+
.merge(
42+
partsupp,
43+
left_on=["L_SUPPKEY", "L_PARTKEY"],
44+
right_on=["PS_SUPPKEY", "PS_PARTKEY"],
4245
)
46+
.merge(supplier, left_on="L_SUPPKEY", right_on="S_SUPPKEY")
4347
.merge(orders, left_on="L_ORDERKEY", right_on="O_ORDERKEY")
4448
.merge(nation, left_on="S_NATIONKEY", right_on="N_NATIONKEY")
4549
)

0 commit comments

Comments
 (0)