From 4829f49f59f17e3bd7bd01a8f9067712d1efed16 Mon Sep 17 00:00:00 2001 From: Trevor Bergeron Date: Fri, 21 Feb 2025 20:22:40 +0000 Subject: [PATCH] chore: Amend tpch q9 to more closely resemble sql --- third_party/bigframes_vendored/tpch/queries/q9.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/third_party/bigframes_vendored/tpch/queries/q9.py b/third_party/bigframes_vendored/tpch/queries/q9.py index 6af33f7569..5c9ca1e9c3 100644 --- a/third_party/bigframes_vendored/tpch/queries/q9.py +++ b/third_party/bigframes_vendored/tpch/queries/q9.py @@ -33,13 +33,17 @@ def q(project_id: str, dataset_id: str, session: bigframes.Session): ) q_final = ( - part.merge(partsupp, left_on="P_PARTKEY", right_on="PS_PARTKEY") - .merge(supplier, left_on="PS_SUPPKEY", right_on="S_SUPPKEY") - .merge( + part.merge( lineitem, - left_on=["P_PARTKEY", "PS_SUPPKEY"], - right_on=["L_PARTKEY", "L_SUPPKEY"], + left_on="P_PARTKEY", + right_on="L_PARTKEY", + ) + .merge( + partsupp, + left_on=["L_SUPPKEY", "L_PARTKEY"], + right_on=["PS_SUPPKEY", "PS_PARTKEY"], ) + .merge(supplier, left_on="L_SUPPKEY", right_on="S_SUPPKEY") .merge(orders, left_on="L_ORDERKEY", right_on="O_ORDERKEY") .merge(nation, left_on="S_NATIONKEY", right_on="N_NATIONKEY") )