Skip to content

Commit 81d3179

Browse files
authored
feat(polars): add ArrayRemove operation (#10322)
1 parent eb14daa commit 81d3179

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

ibis/backends/polars/compiler.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1521,3 +1521,10 @@ def visit_ArrayContains(op, **kw):
15211521
arg = translate(op.arg, **kw)
15221522
value = translate(op.other, **kw)
15231523
return arg.list.contains(value)
1524+
1525+
1526+
@translate.register(ops.ArrayRemove)
1527+
def visit_ArrayRemove(op, **kw):
1528+
arg = translate(op.arg, **kw)
1529+
value = _literal_value(op.other)
1530+
return arg.list.set_difference(pl.lit([value]))

ibis/backends/tests/test_array.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -680,7 +680,6 @@ def test_array_position(con, a, expected_array):
680680

681681

682682
@builtin_array
683-
@pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError)
684683
@pytest.mark.parametrize(
685684
("input", "expected"),
686685
[

0 commit comments

Comments
 (0)