From 6956f34840a7eecae55b882448598647de5ed5f4 Mon Sep 17 00:00:00 2001
From: Nicole00 <anqi.wang@vesoft.com>
Date: Thu, 4 Nov 2021 11:00:09 +0800
Subject: [PATCH] use hasWeight to construct the initial graph

---
 .../vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala b/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala
index e6284e1..b8ba056 100644
--- a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala
+++ b/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala
@@ -30,7 +30,7 @@ object BetweennessCentralityAlgo {
             betweennessConfig: BetweennessConfig,
             hasWeight: Boolean): DataFrame = {
 
-    val graph: Graph[None.type, Double] = NebulaUtil.loadInitGraph(dataset, false)
+    val graph: Graph[None.type, Double] = NebulaUtil.loadInitGraph(dataset, hasWeight)
     val betweennessGraph                = execute(graph, betweennessConfig.maxIter, hasWeight)
 
     val schema = StructType(