From f90d05888b70bf08b19148521c0028e581f80c75 Mon Sep 17 00:00:00 2001 From: Webbley Date: Tue, 17 Mar 2020 18:02:46 +0800 Subject: [PATCH] add num_edge to holder list --- pgl/graph_wrapper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pgl/graph_wrapper.py b/pgl/graph_wrapper.py index 85b9a7b..9545a82 100644 --- a/pgl/graph_wrapper.py +++ b/pgl/graph_wrapper.py @@ -623,6 +623,7 @@ class GraphWrapper(BaseGraphWrapper): self._indegree, self._graph_lod, self._num_graph, + self._num_edges, ]) def __create_graph_node_feat_holders(self, node_feat_name, node_feat_shape, -- GitLab