From c7e144bc01f5d480ec5478a5ca12a5dedc4846f5 Mon Sep 17 00:00:00 2001
From: jeanpouget-abadie <jean.pougetabadie@gmail.com>
Date: Sat, 6 Dec 2014 15:55:22 -0500
Subject: updated greedy algorithm

---
 jpa_test/cascade_creation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'jpa_test/cascade_creation.py')

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
 
-- 
cgit v1.2.3-70-g09d2