diff --git a/epidemic.py b/epidemic.py index 8b74f05..7a12c9c 100644 --- a/epidemic.py +++ b/epidemic.py @@ -160,7 +160,7 @@ class EpidemicSim: self.G.nodes[node][attr] = val def get_state(self, node): - return self.G.node[node]['state'] + return self.G.nodes[node]['state'] def get_people(self): return self.people