diff options
Diffstat (limited to 'jpa_test/cascade_creation.py')
| -rw-r--r-- | jpa_test/cascade_creation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/jpa_test/cascade_creation.py b/jpa_test/cascade_creation.py index dbb9e54..9a26c03 100644 --- a/jpa_test/cascade_creation.py +++ b/jpa_test/cascade_creation.py @@ -65,7 +65,7 @@ class Cascade(list): """ candidate_infectors = collections.Counter() for t in self.infection_time(node): - if t > 0: + if t: # avoid cases where t=0 or t is None candidate_infectors.update(np.where(self[t-1])[0]) return candidate_infectors |