format

ibis-project · Jul 13, 2024 · 3ba384d · 3ba384d
1 parent 9507bdc
commit 3ba384d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/ibis/formats/pyarrow.py b/ibis/formats/pyarrow.py
@@ -101,7 +101,10 @@ def to_ibis(cls, typ: pa.DataType, nullable=True) -> dt.DataType:
             return dt.Map(key_dtype, value_dtype, nullable=nullable)
         elif pa.types.is_dictionary(typ):
             return cls.to_ibis(typ.value_type)
-        elif isinstance(typ, pa.ExtensionType) and type(typ).__name__ == "GeometryExtensionType":
+        elif (
+            isinstance(typ, pa.ExtensionType)
+            and type(typ).__name__ == "GeometryExtensionType"
+        ):
             from geoarrow import types as gat
 
             gat.type_pyarrow.register_extension_types()

diff --git a/ibis/formats/tests/test_pyarrow.py b/ibis/formats/tests/test_pyarrow.py
@@ -186,4 +186,6 @@ def test_unknown_dtype_gets_converted_to_string():
 def test_geo_gets_converted_to_geoarrow(ibis_type):
     type_pyarrow = pytest.importorskip("geoarrow.types.type_pyarrow")
 
-    assert isinstance(PyArrowType.from_ibis(ibis_type), type_pyarrow.GeometryExtensionType)
+    assert isinstance(
+        PyArrowType.from_ibis(ibis_type), type_pyarrow.GeometryExtensionType
+    )