Pyspark dataframe ML model transformations

Question

I\'d need some help with Apache Spark please.

I have already trained an ML model in PySpark which however needs two more columns to run over the test set, these 2 col