From 261c01419ecb66ccb7d45d8010fcdea72922906b Mon Sep 17 00:00:00 2001
From: Parsa744 <51270701+Parsa744@users.noreply.github.com>
Date: Tue, 23 Apr 2024 15:05:45 +0200
Subject: [PATCH] Update network_run.py

---
 network_run.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/network_run.py b/network_run.py
index ceea49f..efef2a7 100644
--- a/network_run.py
+++ b/network_run.py
@@ -31,9 +31,16 @@
 sim_time_test = sim_time
 x_len_test = sim_time_test/dt
 for nn in range(n):
-    if nn < ne: p_rate = b_rate*(1+m_exc*np.cos(2*(th - po_init[nn])))
-    else: p_rate = b_rate*(1+m_inh*np.cos(2*(th - po_init[nn])))
-    rate_ev = np.random.poisson(p_rate*dt/1000., x_len_test).tolist()
+    if nn < ne:
+        p_rate = b_rate*(1+m_exc*np.cos(2*(th - po_init[nn])))
+    else:
+        p_rate = b_rate*(1+m_inh*np.cos(2*(th - po_init[nn])))
+    try:
+        rate_ev = np.random.poisson(lam=int(p_rate * dt / 1000), size=int(x_len_test)).tolist()
+    except TypeError as e:
+        print(f"TypeError: {e}")
+        print(f"p_rate * dt / 1000 evaluated to {p_rate * dt / 1000} which is cast to {int(p_rate * dt / 1000)}")
+        print(f"Size parameter x_len_test is {x_len_test}, should be integer but was cast to {int(x_len_test)}")
     x_bp.append(rate_ev)
 x_bp = np.array(x_bp)
 
@@ -57,7 +64,7 @@
 stim_rng_tot = []
 for blk in range(block_no):
     print(blk)
-    stim_rng = np.random.uniform(0, np.pi, stim_no)
+    stim_rng = np.random.uniform(0, np.pi, int(stim_no))
     stim_rng_tot.append(stim_rng)
     t_stim = sim_time / len(stim_rng)
     x_wp = []
@@ -66,7 +73,10 @@
         for st in stim_rng:
             if nn < ne: p_rate = b_rate*(1+m_exc*np.cos(2*(st - po_init[nn])))
             else: p_rate = b_rate*(1+m_inh*np.cos(2*(st - po_init[nn])))
-            rates = rates + np.random.poisson(p_rate*dt/1000., x_len/len(stim_rng)).tolist()
+            try:
+                rate_ev = np.random.poisson(lam=int(p_rate * dt / 1000), size=int(x_len / len(stim_rng))).tolist()
+            except TypeError as e:
+                print(f"TypeError: {e}")
         rates = np.array(rates)
         x_wp.append(rates)
     x_wp = np.array(x_wp)