From 691975db34353cd75a4b9a7d700e55ae8b135c19 Mon Sep 17 00:00:00 2001 From: seanerice Date: Tue, 28 Apr 2020 20:15:48 -0400 Subject: [PATCH] Fix deprecated networkx calls --- epidemic.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/epidemic.py b/epidemic.py index 5aea9a6..4e2915d 100644 --- a/epidemic.py +++ b/epidemic.py @@ -147,16 +147,16 @@ class EpidemicSim: }}) def increment_time(self, node): - self.G.node[node]['time_infected'] += 1 + self.G.nodes[node]['time_infected'] += 1 def get_attr(self, node, attr): - return self.G.node[node][attr] + return self.G.nodes[node][attr] def get_attrs(self, node): - return self.G.node[node] + return self.G.nodes[node] def set_attr(self, node, attr, val): - self.G.node[node][attr] = val + self.G.nodes[node][attr] = val def get_state(self, node): return self.get_attr(node, 'state') @@ -192,8 +192,8 @@ class EpidemicSim: elif state == 'I': # determine whether they should quarantine if n_attrs['test_submitted']: self.set_attr(n, 'days_since_submitted_test', - n_attrs['days_since_submitted_test'] + 1 - ) + n_attrs['days_since_submitted_test'] + 1 + ) if n_attrs['days_since_submitted_test'] == n_attrs['test_turnaround']: self.update_state(n, 'Q') self.confirmed += 1 -- 2.51.2