fix(polars): handle pl.Array

ibis-project · Oct 1, 2024 · fe322ba · fe322ba
1 parent 41462c5
commit fe322ba
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ibis/formats/polars.py b/ibis/formats/polars.py
@@ -66,7 +66,7 @@ def to_ibis(cls, typ: pl.DataType, nullable=True) -> dt.DataType:
             except AttributeError:  # pragma: no cover
                 time_unit = typ.tu  # pragma: no cover
             return dt.Interval(unit=time_unit, nullable=nullable)
-        elif base_type is pl.List:
+        elif base_type is pl.List or base_type is pl.Array:
             return dt.Array(cls.to_ibis(typ.inner), nullable=nullable)
         elif base_type is pl.Struct:
             return dt.Struct.from_tuples(

diff --git a/ibis/formats/tests/test_polars.py b/ibis/formats/tests/test_polars.py
@@ -171,3 +171,7 @@ def test_convert_table():
     )
     assert df.equals(sol)
     assert df.schema == sol.schema
+
+
+def test_array_type():
+    assert PolarsType.to_ibis(pl.Array(pl.Int64, 2)) == dt.Array(dt.int64)