Improve Switch implementation

mattbit · Sep 30, 2022 · 364781e · 364781e
1 parent 10010ae
commit 364781e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/er/model/network.py b/er/model/network.py
@@ -154,9 +154,12 @@ def open(self, source, time):
         if interval < 0:
             raise Exception("Switch has no memory of previous time!")
 
+        p_switch = (1 - np.exp(-2 * interval / self.timescale)) / 2
+
+        if np.random.rand() <= p_switch:
+            self.status = (self.status + 1) % 2
+
         self.time = time
-        self.status = (
-            self.status + np.random.poisson(interval / self.timescale)) % 2
 
         return not (source == self.source) ^ (self.status)  # xnor