feat(polars): add ArrayIndex operation

Author: IndexSeek

ibis/backends/polars/compiler.py

@@ -1481,3 +1481,10 @@ def execute_date_delta(op, **kw):
     delta = left - right
     method_name = f"total_{_literal_value(op.part)}s"
     return getattr(delta.dt, method_name)()
+
+
+@translate.register(ops.ArrayIndex)
+def execute_array_index(op, **kw):
+    arg = translate(op.arg, **kw)
+    index = translate(op.index, **kw)
+    return arg.list.get(index)

ibis/backends/tests/test_array.py

@@ -204,7 +204,6 @@ def test_np_array_literal(con):
 
 
 @pytest.mark.parametrize("idx", range(3))
-@pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError)
 def test_array_index(con, idx):
     arr = [1, 2, 3]
     expr = ibis.literal(arr)