diff options
| author | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-12-06 15:55:22 -0500 |
|---|---|---|
| committer | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-12-06 15:55:22 -0500 |
| commit | c7e144bc01f5d480ec5478a5ca12a5dedc4846f5 (patch) | |
| tree | 7198c31638f50446de9b33851d0ddb661d7f63eb /jpa_test/cascade_creation.py | |
| parent | 91d31698d5a7eb6e3381f21161658a698362bdf4 (diff) | |
| download | cascades-c7e144bc01f5d480ec5478a5ca12a5dedc4846f5.tar.gz | |
updated greedy algorithm
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 |