feat(polars): implement approx_nunique

ibis-project · Aug 16, 2024 · 3f3738d · 3f3738d
1 parent fb22e20
commit 3f3738d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -708,6 +708,7 @@ def struct_column(op, **kw):
     ops.All: "all",
     ops.Any: "any",
     ops.ApproxMedian: "median",
+    ops.ApproxCountDistinct: "approx_n_unique",
     ops.Count: "count",
     ops.CountDistinct: "n_unique",
     ops.Max: "max",

diff --git a/ibis/backends/tests/test_aggregation.py b/ibis/backends/tests/test_aggregation.py
@@ -478,9 +478,6 @@ def mean_and_std(v):
             lambda t, where: t.string_col.approx_nunique(where=where),
             lambda t, where: t.string_col[where].nunique(),
             id="approx_nunique",
-            marks=[
-                pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError),
-            ],
         ),
         param(
             lambda t, where: t.bigint_col.bit_and(where=where),