Fix polars eager

pola-rs · Apr 15, 2024 · 3323b9f · 3323b9f
1 parent 510ca76
commit 3323b9f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git 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
@@ -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,
     )