change list to set for performance improvement

2dev2 · May 11, 2015 · c55ad5f · c55ad5f
1 parent e653c6e
commit c55ad5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphs/graph.py b/graphs/graph.py
@@ -140,7 +140,7 @@ def get_edge_weights(self):
         unique_list = {}
         for u in self.nodes():
             for v in self.neighbors(u):
-                if u not in unique_list.get(v, []):
+                if u not in unique_list.get(v, set()):
                     edge_list.append((self.node_neighbors[u][v], (u, v)))
-                    unique_list.setdefault(u, []).append(v)
+                    unique_list.setdefault(u, set()).add(v)
         return edge_list