Clickbench

Author: gatesn

pyvortex/clickbench.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# ruff: noqa: E501
 import argparse
 import json
 import os.path

pyvortex/python/vortex/_lib/expr.pyi

@@ -5,8 +5,8 @@ import vortex.expr as ve
 
 @final
 class Expr:
-    def __eq__(self, other: ve.IntoExpr) -> Expr: ...
-    def __ne__(self, other: ve.IntoExpr) -> Expr: ...
+    def __eq__(self, other: ve.IntoExpr) -> Expr: ...  # type: ignore[override]
+    def __ne__(self, other: ve.IntoExpr) -> Expr: ...  # type: ignore[override]
     def __lt__(self, other: ve.IntoExpr) -> Expr: ...
     def __le__(self, other: ve.IntoExpr) -> Expr: ...
     def __gt__(self, other: ve.IntoExpr) -> Expr: ...

pyvortex/python/vortex/polars_.py

@@ -98,7 +98,7 @@ def _polars_to_vortex(expr: dict) -> ve.Expr:
 
     if "Function" in expr:
         expr = expr["Function"]
-        inputs = [_polars_to_vortex(e) for e in expr["input"]]
+        _inputs = [_polars_to_vortex(e) for e in expr["input"]]
 
         fn = expr["function"]
         if "Boolean" in fn:

pyvortex/test/test_polars_.py

@@ -1,5 +1,3 @@
-from collections.abc import Iterator
-
 import polars as pl
 import pytest
 
@@ -33,19 +31,3 @@
 def test_exprs(polars: pl.Expr, vortex: ve.Expr):
     # Dump the clickbench filters
     assert polars_to_vortex(polars) == vortex
-
-
-def test_io_source():
-    import polars as pl
-    from polars.io.plugins import register_io_source
-
-    def _io_source(
-        with_columns: list[str] | None,
-        predicate: pl.Expr | None,
-        n_rows: int | None,
-        batch_size: int | None,
-    ) -> Iterator[pl.DataFrame]:
-        yield from []
-
-    df = register_io_source(_io_source, schema=pl.Schema([("a", pl.Int64)]))
-    df.collect()