made transpose lazy

freeman-lab · Jan 30, 2015 · 33e9219 · 33e9219
1 parent 5a274cd
commit 33e9219
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -238,7 +238,7 @@ class BlockMatrix(
     val transposedBlocks = blocks.map { case ((blockRowIndex, blockColIndex), mat) =>
       ((blockColIndex, blockRowIndex), mat.transpose)
     }
-    new BlockMatrix(transposedBlocks, colsPerBlock, rowsPerBlock, numCols(), numRows())
+    new BlockMatrix(transposedBlocks, colsPerBlock, rowsPerBlock, nCols, nRows)
   }
 
   /** Collects data and assembles a local dense breeze matrix (for test only). */