diff --git a/queries/duckdb/q11.py b/queries/duckdb/q11.py index 2e8533c..03f8b12 100644 --- a/queries/duckdb/q11.py +++ b/queries/duckdb/q11.py @@ -1,6 +1,9 @@ import duckdb from queries.duckdb import utils +from settings import Settings + +settings = Settings() Q_NUM = 11 @@ -9,6 +12,8 @@ def q() -> None: supplier_ds = utils.get_supplier_ds() part_supp_ds = utils.get_part_supp_ds() nation_ds = utils.get_nation_ds() + scale_factor = settings.scale_factor + fraction = 0.0001 / scale_factor query_str = f""" select @@ -26,7 +31,7 @@ def q() -> None: ps_partkey having sum(ps_supplycost * ps_availqty) > ( select - sum(ps_supplycost * ps_availqty) * 0.0001 + sum(ps_supplycost * ps_availqty) * {fraction} from {part_supp_ds}, {supplier_ds}, diff --git a/queries/polars/q11.py b/queries/polars/q11.py index c79b86b..f9dd54a 100644 --- a/queries/polars/q11.py +++ b/queries/polars/q11.py @@ -3,6 +3,9 @@ import polars as pl from queries.polars import utils +from settings import Settings + +settings = Settings() Q_NUM = 11 @@ -23,7 +26,7 @@ def q( assert supplier is not None var1 = "GERMANY" - var2 = 0.0001 + var2 = 0.0001 / settings.scale_factor q1 = ( partsupp.join(supplier, left_on="ps_suppkey", right_on="s_suppkey") diff --git a/queries/pyspark/q11.py b/queries/pyspark/q11.py index c08b9c4..1c72c5a 100644 --- a/queries/pyspark/q11.py +++ b/queries/pyspark/q11.py @@ -1,10 +1,16 @@ from queries.pyspark import utils +from settings import Settings + +settings = Settings() Q_NUM = 11 def q() -> None: - query_str = """ + scale_factor = settings.scale_factor + fraction = 0.0001 / scale_factor + + query_str = f""" select ps_partkey, round(sum(ps_supplycost * ps_availqty), 2) as value @@ -20,7 +26,7 @@ def q() -> None: ps_partkey having sum(ps_supplycost * ps_availqty) > ( select - sum(ps_supplycost * ps_availqty) * 0.0001 + sum(ps_supplycost * ps_availqty) * {fraction} from partsupp, supplier,