Skip to content

Commit 4b9803b

Browse files
committed
[SPARK-8563] [MLlib] Fixed a build error.
1 parent c2ccd89 commit 4b9803b

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ class IndexedRowMatrix(
146146
val indexedRows = indices.zip(svd.U.rows).map { case (i, v) =>
147147
IndexedRow(i, v)
148148
}
149-
new IndexedRowMatrix(indexedRows, svd.U.numRows(), svd.U.numCols())
149+
new IndexedRowMatrix(indexedRows, svd.U.numRows(), svd.U.numCols().toInt)
150150
} else {
151151
null
152152
}

mllib/src/test/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrixSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,8 +142,8 @@ class IndexedRowMatrixSuite extends SparkFunSuite with MLlibTestSparkContext {
142142
assert(svd.U.numRows() === m)
143143
assert(svd.U.numCols() === k)
144144
assert(svd.s.size === k)
145-
assert(svd.V.numRows() === n)
146-
assert(svd.V.numCols() === k)
145+
assert(svd.V.numRows === n)
146+
assert(svd.V.numCols === k)
147147
}
148148

149149
test("validate k in svd") {

0 commit comments

Comments
 (0)