fixed Python ruff errors

appian · Jun 13, 2024 · eed8c96 · eed8c96
1 parent 0241501
commit eed8c96
Show file tree

Hide file tree

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