cleaned up raf-cell

NACLab · Aug 9, 2024 · 73e5aa1 · 73e5aa1
1 parent 94f37f7
commit 73e5aa1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ngclearn/components/neurons/spiking/RAFCell.py b/ngclearn/components/neurons/spiking/RAFCell.py
@@ -159,14 +159,16 @@ def _advance_state(t, dt, tau_v, resist_v, tau_w, thr, omega, b,
         ## continue with centered dynamics
         j_ = j * resist_v
         if intgFlag == 1:  ## RK-2/midpoint
+            ## Note: we integrate ODEs in order: first w, then v
             w_params = (j_, v, tau_w, omega, b)
             _, _w = step_rk2(0., w, _dfw, dt, w_params)
-            v_params = (j_, w, tau_v, omega, b)
+            v_params = (j_, _w, tau_v, omega, b)
             _, _v = step_rk2(0., v, _dfv, dt, v_params)
         else:  # integType == 0 (default -- Euler)
+            ## Note: we integrate ODEs in order: first w, then v
             w_params = (j_, v, tau_w, omega, b)
             _, _w = step_euler(0., w, _dfw, dt, w_params)
-            v_params = (j_, w, tau_v, omega, b)
+            v_params = (j_, _w, tau_v, omega, b)
             _, _v = step_euler(0., v, _dfv, dt, v_params)
         s = _emit_spike(_v, thr)
         ## hyperpolarize/reset/snap variables