diff --git a/network_attack_propagation.py b/network_attack_propagation.py index ddad01c..d49d59f 100644 --- a/network_attack_propagation.py +++ b/network_attack_propagation.py @@ -188,7 +188,8 @@ def main(): do_graph_animation('ring.gif', ring, 20, nx.circular_layout) star, star_avg_rank = star_network() - do_graph_animation('star.gif', star, 5, nx.kamada_kawai_layout) + list(star.get_nodes())[1].is_infected = True + do_graph_animation('star.gif', star, 20, nx.kamada_kawai_layout) if __name__ == "__main__":