feat(polars): add ArrayRemove operation (#10322)

ibis-project · Oct 16, 2024 · 81d3179 · 81d3179
1 parent eb14daa
commit 81d3179
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -1521,3 +1521,10 @@ def visit_ArrayContains(op, **kw):
     arg = translate(op.arg, **kw)
     value = translate(op.other, **kw)
     return arg.list.contains(value)
+
+
+@translate.register(ops.ArrayRemove)
+def visit_ArrayRemove(op, **kw):
+    arg = translate(op.arg, **kw)
+    value = _literal_value(op.other)
+    return arg.list.set_difference(pl.lit([value]))
diff --git a/ibis/backends/tests/test_array.py b/ibis/backends/tests/test_array.py
@@ -680,7 +680,6 @@ def test_array_position(con, a, expected_array):
 
 
 @builtin_array
-@pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError)
 @pytest.mark.parametrize(
     ("input", "expected"),
     [