GraphWave Self Loop fix.

benedekrozemberczki · Feb 13, 2020 · 7ebcf50 · 7ebcf50
1 parent 8a713c0
commit 7ebcf50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/karateclub/node_embedding/structural/graphwave.py b/karateclub/node_embedding/structural/graphwave.py
@@ -118,9 +118,9 @@ def fit(self, graph):
         Arg types:
             * **graph** *(NetworkX graph)* - The graph to be embedded.
         """
+        graph.remove_edges_from(nx.selfloop_edges(graph))
         self._create_evaluation_points()
         self._check_size(graph)
-        graph = graph.remove_edges_from(nx.selfloop_edges(graph))
         self.G = pygsp.graphs.Graph(nx.adjacency_matrix(graph))
 
         if self.mechanism == "exact":