✨ solve DegreeStatic algorithm config problem (#77)

vesoft-inc · Aug 4, 2023 · 3fc5955 · 3fc5955
1 parent 397a8a3
commit 3fc5955
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/Main.scala b/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/Main.scala
@@ -22,7 +22,8 @@ import com.vesoft.nebula.algorithm.config.{
   Node2vecConfig,
   PRConfig,
   ShortestPathConfig,
-  SparkConfig
+  SparkConfig,
+  DegreeStaticConfig
 }
 import com.vesoft.nebula.algorithm.lib.{
   BetweennessCentralityAlgo,
@@ -171,7 +172,8 @@ object Main {
           ShortestPathAlgo(spark, dataSet, spConfig, hasWeight)
         }
         case "degreestatic" => {
-          DegreeStaticAlgo(spark, dataSet)
+          val dsConfig = DegreeStaticConfig.getDegreeStaticConfig(configs)
+          DegreeStaticAlgo(spark, dataSet, dsConfig)
         }
         case "kcore" => {
           val kCoreConfig = KCoreConfig.getKCoreConfig(configs)