diff --git a/tdigest_plugin/__init__.py b/tdigest_plugin/__init__.py index 2c4ff89..1e990e5 100644 --- a/tdigest_plugin/__init__.py +++ b/tdigest_plugin/__init__.py @@ -5,7 +5,8 @@ import polars as pl from polars.plugins import register_plugin_function -from tdigest_plugin.utils import parse_into_expr, parse_version + +from tdigest_plugin.utils import parse_version if TYPE_CHECKING: from polars.type_aliases import IntoExpr diff --git a/tests/test_estimate_quantiles.py b/tests/test_estimate_quantiles.py index fedcb3d..5b50a40 100644 --- a/tests/test_estimate_quantiles.py +++ b/tests/test_estimate_quantiles.py @@ -1,8 +1,8 @@ -from decimal import InvalidOperation import polars as pl import pytest -from polars import Float32, Float64, Int16, Int32, Utf8, col +from polars import Float32, Int32, Utf8, col + from tdigest_plugin import estimate_median, estimate_quantile, tdigest df_int = pl.DataFrame( @@ -102,7 +102,7 @@ def test_estimate_quantile_f32(): def test_estimate_quantile_utf8(): with pytest.raises(pl.exceptions.ComputeError): - df_median = ( + ( df_float.with_columns(col("values").cast(Utf8)) .group_by("group") .agg(