[SPARK-3974] Change return type of cache and persist

marmbrus · Jan 20, 2015 · 1e8bb2a · 1e8bb2a
1 parent 239ab4b
commit 1e8bb2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/BlockMatrix.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/BlockMatrix.scala
@@ -172,13 +172,13 @@ class BlockMatrix(
   }
 
   /** Cache the underlying RDD. */
-  def cache(): DistributedMatrix = {
+  def cache(): BlockMatrix = {
     rdd.cache()
     this
   }
 
   /** Set the storage level for the underlying RDD. */
-  def persist(storageLevel: StorageLevel): DistributedMatrix = {
+  def persist(storageLevel: StorageLevel): BlockMatrix = {
     rdd.persist(storageLevel)
     this
   }