pyspark 3.0 - PythonUDF Runner error whilst applying OneVsRest model

Question

I\'ve been able to train a OneVsRest model, and save it.The model has 100 classes. I\'ve managed to load the model, and currently trying to apply the model on some test data