Is it currently possible (or planned) to support output of multiple rows, or multiple columns from a python UDF?

is something Similar to the Flatmap java udf possible with python UDFs?

Unfortunately, this is currently not possible, but you should be able to approximate the same behavior by writing a python function that returns a list and exploding that into multiple rows with Builder’s built-in transforms.

Is that approach sufficient for your use case?

Hey Charlie, thanks for the info! We will use the built in PB transforms for now.