This is an automated email from the ASF dual-hosted git repository.

imbruced pushed a commit to branch arrow-worker
in repository https://gitbox.apache.org/repos/asf/sedona.git


The following commit(s) were added to refs/heads/arrow-worker by this push:
     new bd50e743f8 add sedonadb sedona udf worker example
bd50e743f8 is described below

commit bd50e743f863d7f833fe48afa8dbb11913dfab8a
Author: pawelkocinski <[email protected]>
AuthorDate: Sun Jan 4 23:27:06 2026 +0100

    add sedonadb sedona udf worker example
---
 python/sedona/spark/worker/worker.py                | 2 --
 python/tests/utils/test_sedona_db_vectorized_udf.py | 3 ++-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/sedona/spark/worker/worker.py 
b/python/sedona/spark/worker/worker.py
index 571134f407..6b2a18c8f2 100644
--- a/python/sedona/spark/worker/worker.py
+++ b/python/sedona/spark/worker/worker.py
@@ -170,8 +170,6 @@ def register_sedona_db_udf(infile, pickle_ser) -> UDFInfo:
     udf = None
     for _ in range(num_udfs):
         udf = read_udf(infile, pickle_ser)
-        # Here we would register the UDF with SedonaDB's internal context
-
 
     return udf
 
diff --git a/python/tests/utils/test_sedona_db_vectorized_udf.py 
b/python/tests/utils/test_sedona_db_vectorized_udf.py
index 749d45420e..6021811e91 100644
--- a/python/tests/utils/test_sedona_db_vectorized_udf.py
+++ b/python/tests/utils/test_sedona_db_vectorized_udf.py
@@ -5,6 +5,7 @@ import shapely
 from sedona.sql import GeometryType
 from pyspark.sql.functions import expr, lit
 from pyspark.sql.types import DoubleType, IntegerType
+from sedona.spark.sql import ST_X
 
 
 class TestSedonaDBArrowFunction(TestBase):
@@ -28,7 +29,7 @@ class TestSedonaDBArrowFunction(TestBase):
             ["id", "wkt"],
         ).withColumn("wkt", expr("ST_GeomFromWKT(wkt)"))
 
-        df.select(my_own_function(df.wkt, lit(100)).alias("geom")).show()
+        df.select(ST_X(my_own_function(df.wkt, lit(100)).alias("geom"))).show()
 
     def test_geometry_to_double(self):
         @sedona_db_vectorized_udf(return_type=DoubleType(), 
input_types=[GeometryType()])

Reply via email to