diff options
| -rw-r--r-- | graphx/src/main/scala/org/apache/spark/graphx/util/GraphGenerators.scala | 9 |
1 files changed, 1 insertions, 8 deletions
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/util/GraphGenerators.scala b/graphx/src/main/scala/org/apache/spark/graphx/util/GraphGenerators.scala index dbea233c34..57422ce3f1 100644 --- a/graphx/src/main/scala/org/apache/spark/graphx/util/GraphGenerators.scala +++ b/graphx/src/main/scala/org/apache/spark/graphx/util/GraphGenerators.scala @@ -44,14 +44,7 @@ object GraphGenerators { def generateRandomEdges(src: Int, numEdges: Int, maxVertexID: Int): Array[Edge[Int]] = { val rand = new Random() - var dsts: Set[Int] = Set() - while (dsts.size < numEdges) { - val nextDst = rand.nextInt(maxVertexID) - if (nextDst != src) { - dsts += nextDst - } - } - dsts.map(dst => Edge[Int](src, dst, 1)).toArray + Array.fill(maxVertexID) { Edge[Int](src, rand.nextInt(maxVertexID), 1) } } /** |