From fd4b442d8e429c97f2d72929f5c211f5f5e4d2b3 Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Fri, 16 Oct 2015 16:52:17 -0400
Subject: Fix source selection + cosmetic improvement for Jean
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 simulation/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'simulation/main.py')

diff --git a/simulation/main.py b/simulation/main.py
index 63c8c02..b7b3eaa 100644
--- a/simulation/main.py
+++ b/simulation/main.py
@@ -39,8 +39,8 @@ def simulate_cascade(x, graph):
 
 def simulate_cascades(n, graph):
     for _ in xrange(n):
-        x = np.zeros(g.shape[0]).astype(bool)
-        x[nr.random()] = True
+        x = np.zeros(g.shape[0], dtype=bool)
+        x[nr.randint(0, g.shape[0])] = True
         yield simulate_cascade(x, graph)
 
 
-- 
cgit v1.2.3-70-g09d2