update template query

Author: RobRossmiller-TG

GDBMS_ALGO/centrality/weighted_degree_cent.gsql

@@ -1,5 +1,4 @@
 CREATE TEMPLATE QUERY GDBMS_ALGO.centrality.weighted_degree_cent(STRING v_type, STRING e_type, STRING reverse_e_type, string weight_attribute, BOOL in_degree = TRUE, BOOL out_degree = TRUE, INT top_k=100, BOOL print_results = TRUE, STRING result_attribute = "",STRING file_path = "") SYNTAX V1 {
-
    /*
    First Author: <First Author Name>
    First Commit Date:  <First Commit Date>
@@ -22,18 +21,20 @@ CREATE TEMPLATE QUERY GDBMS_ALGO.centrality.weighted_degree_cent(STRING v_type,
       for undirected graph, you only need to set e_type and in_degree
 
    Publications:
-      <link>
+      NA
 
    TigerGraph Documentation:
-      <link>
+      https://docs.tigergraph.com/graph-ml/current/centrality-algorithms/weighted-degree-centrality
 
    Parameters:
       v_type:
-         vertex types to traverse
+         Vertex types to traverse
       e_type:
-         edge types to traverse
+         Edge types to traverse
       reverse_e_type:
-         for indegree use
+         For indegree use
+      weight_attribute:
+         The edge weight attribute name
       in_degree:
          If True, count incoming relationships
       out_degree:
@@ -43,27 +44,28 @@ CREATE TEMPLATE QUERY GDBMS_ALGO.centrality.weighted_degree_cent(STRING v_type,
       print_results:
          If True, print the results
       result_attribute:
-         attribute to write result to
+         Attribute to write result to
       file_path:
-         file to write CSV output to
-  */
+         File to write CSV output to
+   */
 
   TYPEDEF TUPLE<VERTEX Vertex_ID, FLOAT score> Vertex_Score;
   HeapAccum<Vertex_Score>(top_k, score DESC) @@top_scores_heap;
-  SumAccum<INT> @sum_degree_score;
+  SumAccum<FLOAT> @sum_degree_score;
   FILE f (file_path);
 
   all = {v_type};
   IF in_degree THEN 
       sll = SELECT s 
-            FROM all:s-(reverse_e_type:e)-:t
-            ACCUM s.@sum_degree_score+=e.getAttr(weight_attribute,"INT");
+            FROM all:s -(reverse_e_type:e)- :t
+            ACCUM s.@sum_degree_score += e.getAttr(weight_attribute, "FLOAT");
   END;
   IF out_degree THEN 
       sll = SELECT s 
-            FROM all:s-(e_type:e)-:t
-            ACCUM s.@sum_degree_score+=e.getAttr(weight_attribute,"INT");
+            FROM all:s -(e_type:e)- :t
+            ACCUM s.@sum_degree_score += e.getAttr(weight_attribute, "FLOAT");
   END;
+
   #Output
   IF file_path != "" THEN
       f.println("Vertex_ID", "Degree");
@@ -72,17 +74,17 @@ CREATE TEMPLATE QUERY GDBMS_ALGO.centrality.weighted_degree_cent(STRING v_type,
   Start = SELECT s 
           FROM all:s
           POST-ACCUM
-          IF result_attribute != "" THEN 
-             s.setAttr(result_attribute, s.@sum_degree_score) 
-          END,
-    
-          IF print_results THEN 
-             @@top_scores_heap += Vertex_Score(s, s.@sum_degree_score) 
-          END,
-    
-          IF file_path != "" THEN 
-             f.println(s, s.@sum_degree_score) 
-          END;
+             IF result_attribute != "" THEN 
+                s.setAttr(result_attribute, s.@sum_degree_score) 
+             END,
+       
+             IF print_results THEN 
+                @@top_scores_heap += Vertex_Score(s, s.@sum_degree_score) 
+             END,
+       
+             IF file_path != "" THEN 
+                f.println(s, s.@sum_degree_score) 
+             END;
 
    IF print_results THEN
        PRINT @@top_scores_heap AS top_scores;