From 3323b9fb84267d89b7f858cac6962100c7c9e939 Mon Sep 17 00:00:00 2001 From: Stijn de Gooijer Date: Mon, 15 Apr 2024 12:24:01 +0200 Subject: [PATCH] Fix polars eager --- queries/common_utils.py | 3 ++- queries/polars/utils.py | 6 +++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/queries/common_utils.py b/queries/common_utils.py index af6a1da..f392f5d 100644 --- a/queries/common_utils.py +++ b/queries/common_utils.py @@ -100,6 +100,7 @@ def run_query_generic( query: Callable[..., Any], query_number: int, library_name: str, + library_version: str | None = None, query_checker: Callable[..., None] | None = None, ) -> None: """Execute a query.""" @@ -109,7 +110,7 @@ def run_query_generic( if settings.run.log_timings: log_query_timing( solution=library_name, - version=version(library_name), + version=library_version or version(library_name), query_number=query_number, time=timer.took, ) diff --git a/queries/polars/utils.py b/queries/polars/utils.py index 18815fd..4ddbac0 100644 --- a/queries/polars/utils.py +++ b/queries/polars/utils.py @@ -70,5 +70,9 @@ def run_query(query_number: int, lf: pl.LazyFrame) -> None: library_name = "polars" if not eager else "polars-eager" run_query_generic( - query, query_number, library_name, query_checker=check_query_result_pl + query, + query_number, + library_name, + library_version=pl.__version__, + query_checker=check_query_result_pl, )