From d565e17b2f7d87ef9129069b798a4b9f4e8a9612 Mon Sep 17 00:00:00 2001 From: Guoqiang Li Date: Sun, 7 Aug 2016 11:08:31 +0800 Subject: [PATCH] Fixing ArrayIndexOutOfBoundsException in DBHPartitioner --- .../com/github/cloudml/zen/ml/partitioner/DBHPartitioner.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml/src/main/scala/com/github/cloudml/zen/ml/partitioner/DBHPartitioner.scala b/ml/src/main/scala/com/github/cloudml/zen/ml/partitioner/DBHPartitioner.scala index 6bd01dde..974c69af 100644 --- a/ml/src/main/scala/com/github/cloudml/zen/ml/partitioner/DBHPartitioner.scala +++ b/ml/src/main/scala/com/github/cloudml/zen/ml/partitioner/DBHPartitioner.scala @@ -66,7 +66,7 @@ object DBHPartitioner { val conf = edges.context.getConf val numPartitions = conf.getInt(cs_numPartitions, edges.partitions.length) val dbh = new DBHPartitioner(numPartitions, 0) - val degGraph = GraphImpl(input.degrees, edges) + val degGraph = GraphImpl(input.degrees, edges, 0, StorageLevel.NONE, StorageLevel.NONE) val newEdges = degGraph.triplets.mapPartitions(_.map(et => (dbh.getKey(et), Edge(et.srcId, et.dstId, et.attr)) )).partitionBy(dbh).map(_._2)