FIX: assign value for dag root when GSL starts from edge (#267)

alibaba · Jul 6, 2023 · 02d8963 · 02d8963
1 parent 3e7f1dd
commit 02d8963
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/graphlearn/python/graph.py b/graphlearn/python/graph.py
@@ -620,6 +620,7 @@ def E(self,
       dag, op_name="GetEdges", params=params)
     source_node.set_output_field(pywrap.kEdgeIds)
     source_node.set_path(edge_type, pywrap.NodeFrom.NODE)
+    dag.root = source_node
 
     # Add sink node to dag
     gsl.SinkNode(dag)