diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/cascade_creation.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/cascade_creation.py b/src/cascade_creation.py index c647b9a..c297f76 100644 --- a/src/cascade_creation.py +++ b/src/cascade_creation.py @@ -25,8 +25,7 @@ class InfluenceGraph(nx.DiGraph): p : proba of rewiring edge """ G = nx.connected_watts_strogatz_graph(n, k, p) - self.add_edges_from(G.edges()) - #TODO: Make the graph directed! + G = nx.convert_to_directed(G) self.add_edges_from(G.edges()) def import_from_file(self, file_name): |