Using sklearn model into pyspark for predicts

Question

I have the pyspark Dataframe and trained sklearn model. I want to use it for predict using MapPartitions method from pyspark.

How can I write wrapper around my model