From 1f0a35505d6f41dadb607eb97edf3ec4ea4662ab Mon Sep 17 00:00:00 2001 From: Neville Li Date: Mon, 7 Jul 2014 15:06:26 -0400 Subject: [PATCH] [SPARK-1977][MLLIB] register mutable BitSet in MovieLenseALS --- .../scala/org/apache/spark/examples/mllib/MovieLensALS.scala | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala b/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala index 28e201d279f41..98aaedb9d7dc9 100644 --- a/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala +++ b/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala @@ -17,6 +17,8 @@ package org.apache.spark.examples.mllib +import scala.collection.mutable + import com.esotericsoftware.kryo.Kryo import org.apache.log4j.{Level, Logger} import scopt.OptionParser @@ -41,6 +43,7 @@ object MovieLensALS { class ALSRegistrator extends KryoRegistrator { override def registerClasses(kryo: Kryo) { kryo.register(classOf[Rating]) + kryo.register(classOf[mutable.BitSet]) } }