fix(geospatial): pretty print data in interactive mode

ibis-project · Dec 5, 2023 · afb04ed · afb04ed
1 parent 724859d
commit afb04ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ibis/expr/types/pretty.py b/ibis/expr/types/pretty.py
@@ -41,6 +41,13 @@ def _(dtype, values):
     return _format_nested([None if v is None else dict(v) for v in values])
 
 
+@format_values.register(dt.GeoSpatial)
+def _(dtype, values):
+    import shapely
+
+    return _format_nested([None if v is None else shapely.from_wkb(v) for v in values])
+
+
 @format_values.register(dt.JSON)
 def _(dtype, values):
     def try_json(v):

diff --git a/ibis/formats/pyarrow.py b/ibis/formats/pyarrow.py
@@ -90,6 +90,7 @@ def __arrow_ext_scalar_class__(self):
     dt.Unknown: pa.string(),
     dt.MACADDR: pa.string(),
     dt.INET: pa.string(),
+    dt.GeoSpatial: pa.binary(),
 }