Fix importing GeometryType from sedona.spark

apache · Jul 31, 2024 · 2c1ac36 · 2c1ac36
1 parent 9ebdd5d
commit 2c1ac36
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/sedona/sql/st_functions.py b/python/sedona/sql/st_functions.py
@@ -2050,4 +2050,4 @@ def ST_Rotate(geometry: ColumnOrName, angle: Union[ColumnOrName, float], originX
 
 # Automatically populate __all__
 __all__ = [name for name, obj in inspect.getmembers(sys.modules[__name__])
-           if inspect.isfunction(obj)]
+           if inspect.isfunction(obj) and name != 'GeometryType']
diff --git a/python/tests/sql/test_st_function_imports.py b/python/tests/sql/test_st_function_imports.py
@@ -33,3 +33,8 @@ def test_import(self):
         ST_Point
         ST_Contains
         ST_Envelope_Aggr
+
+    def test_geometry_type_should_be_a_sql_type(self):
+        from sedona.spark import GeometryType
+        from pyspark.sql.types import UserDefinedType
+        assert isinstance(GeometryType(), UserDefinedType)