Skip to content

Commit 5cd0cf2

Browse files
authored
chore: update q11 (#1359)
1 parent aafb5be commit 5cd0cf2

File tree

1 file changed

+6
-6
lines changed
  • third_party/bigframes_vendored/tpch/queries

1 file changed

+6
-6
lines changed

third_party/bigframes_vendored/tpch/queries/q11.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,20 +18,20 @@ def q(project_id: str, dataset_id: str, session: bigframes.Session):
1818
index_col=bigframes.enums.DefaultIndexKind.NULL,
1919
)
2020

21-
merged_df = partsupp.merge(supplier, left_on="PS_SUPPKEY", right_on="S_SUPPKEY")
22-
merged_df = merged_df.merge(nation, left_on="S_NATIONKEY", right_on="N_NATIONKEY")
21+
nation = nation[nation["N_NAME"] == "GERMANY"]
2322

24-
filtered_df = merged_df[merged_df["N_NAME"] == "GERMANY"]
23+
merged_df = nation.merge(supplier, left_on="N_NATIONKEY", right_on="S_NATIONKEY")
24+
merged_df = merged_df.merge(partsupp, left_on="S_SUPPKEY", right_on="PS_SUPPKEY")
2525

26-
filtered_df["VALUE"] = filtered_df["PS_SUPPLYCOST"] * filtered_df["PS_AVAILQTY"]
27-
grouped = filtered_df.groupby("PS_PARTKEY", as_index=False).agg(
26+
merged_df["VALUE"] = merged_df["PS_SUPPLYCOST"] * merged_df["PS_AVAILQTY"]
27+
grouped = merged_df.groupby("PS_PARTKEY", as_index=False).agg(
2828
VALUE=bpd.NamedAgg(column="VALUE", aggfunc="sum")
2929
)
3030

3131
grouped["VALUE"] = grouped["VALUE"].round(2)
3232

3333
total_value = (
34-
(filtered_df["PS_SUPPLYCOST"] * filtered_df["PS_AVAILQTY"]).to_frame().sum()
34+
(merged_df["PS_SUPPLYCOST"] * merged_df["PS_AVAILQTY"]).to_frame().sum()
3535
)
3636
threshold = (total_value * 0.0001).rename("THRESHOLD")
3737

0 commit comments

Comments
 (0)