[SPARK-1977][MLLIB] register mutable BitSet in MovieLenseALS

nevillelyh · nevillelyh · commit 1f0a35505d6f · 2014-07-07T17:17:41.000-04:00
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
@@ -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])
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,8 @@`
`17`	`17`
`18`	`18`	`package org.apache.spark.examples.mllib`
`19`	`19`
	`20`	`+import scala.collection.mutable`
	`21`	`+`
`20`	`22`	`import com.esotericsoftware.kryo.Kryo`
`21`	`23`	`import org.apache.log4j.{Level, Logger}`
`22`	`24`	`import scopt.OptionParser`
`@@ -41,6 +43,7 @@ object MovieLensALS {`
`41`	`43`	`class ALSRegistrator extends KryoRegistrator {`
`42`	`44`	`override def registerClasses(kryo: Kryo) {`
`43`	`45`	`kryo.register(classOf[Rating])`
	`46`	`+ kryo.register(classOf[mutable.BitSet])`
`44`	`47`	`}`
`45`	`48`	`}`
`46`	`49`