Sure you can reference the commented out code here which builds the signature from a pyspark dataframe with array types. This signature would then be passed in on L275 (also commented out). Please let me know if you get this  working without making inference unbearably slow.