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()])