From cdefcc8c16e96259b16a496e8a9f266dc8956c63 Mon Sep 17 00:00:00 2001 From: Stijn de Gooijer Date: Fri, 5 Apr 2024 14:28:32 +0200 Subject: [PATCH] driveby --- queries/polars/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/queries/polars/utils.py b/queries/polars/utils.py index 847989e..de63450 100644 --- a/queries/polars/utils.py +++ b/queries/polars/utils.py @@ -57,15 +57,15 @@ def get_part_supp_ds() -> pl.LazyFrame: return _scan_ds(settings.dataset_base_dir / "partsupp") -def run_query(q_num: int, lp: pl.LazyFrame) -> None: +def run_query(q_num: int, lf: pl.LazyFrame) -> None: @linetimer(name=f"Overall execution of polars Query {q_num}", unit="s") # type: ignore[misc] def query() -> None: if settings.run.polars_show_plan: - print(lp.explain()) + print(lf.explain()) with CodeTimer(name=f"Get result of polars Query {q_num}", unit="s"): t0 = timeit.default_timer() - result = lp.collect(streaming=settings.run.polars_streaming) + result = lf.collect(streaming=settings.run.polars_streaming) secs = timeit.default_timer() - t0