I've been struggling myslef with it, but after installing pypmml for spark, I can use the other library, maybe it will work for you:

• runtime 10.4 LTS ML
• install pypmml-spark (https://github.com/autodeployai/pypmml-spark)
• install pmml4s-spark (org.pmml4s:pmml4s-spark_2.12:0.9.16, pyspark is eventually using it, https://github.com/autodeployai/pmml4s-spark)

both pyspark & scala works