package demo udf callable for remote execution

This commit is contained in:
Xuanwo
2026-08-21 22:33:20 +08:00
parent b842e87912
commit fc9ec60b69
+5 -1
View File
@@ -741,7 +741,11 @@ def _package_source(function: Callable[..., Any]) -> bytes:
if globals_source:
parts.extend(["", *globals_source])
parts.extend(["", function_source, ""])
return "\n".join(parts).encode("utf-8")
try:
import cloudpickle
except ImportError as error:
raise RuntimeError("@udf remote registration requires cloudpickle") from error
return cloudpickle.dumps(function)
class UdfDefinition: