keep e, i in range [0,1] (#233)

* keep e, i in range [0,1] * refactor as suggested
neurolib-dev · May 2, 2023 · e1e0509 · e1e0509
1 parent 6771599
commit e1e0509
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/neurolib/models/wc/timeIntegration.py b/neurolib/models/wc/timeIntegration.py
@@ -241,6 +241,17 @@ def S_I(x):
             excs[no, i] = excs[no, i - 1] + dt * exc_rhs
             inhs[no, i] = inhs[no, i - 1] + dt * inh_rhs
 
+            # make sure e and i variables do not exceed 1 (can only happen with noise)
+            if excs[no, i] > 1.0:
+                excs[no, i] = 1.0
+            if excs[no, i] < 0.0:
+                excs[no, i] = 0.0
+
+            if inhs[no, i] > 1.0:
+                inhs[no, i] = 1.0
+            if inhs[no, i] < 0.0:
+                inhs[no, i] = 0.0
+
             # Ornstein-Uhlenbeck process
             exc_ou[no] = (
                 exc_ou[no] + (exc_ou_mean - exc_ou[no]) * dt / tau_ou + sigma_ou * sqrt_dt * noise_exc[no]