diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala index fa543cc0dd82c..eb85791e9dbf3 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala @@ -105,7 +105,7 @@ private[ml] object OneVsRestParams extends ClassifierTypeTrait { DefaultParamsWriter.saveMetadata(instance, path, spark, extraMetadata, Some(jsonParams)) val classifierPath = new Path(path, "classifier").toString - instance.getClassifier.asInstanceOf[MLWritable].save(classifierPath) + instance.getClassifier.asInstanceOf[MLWritable].write.session(spark).save(classifierPath) } def loadImpl(