mod to gated-hebb

Author: ago109

Diff for: ngclearn/components/synapses/hebbian/gatedHebbianSynapse.py

@@ -38,14 +38,15 @@ def _compute_update(w_bound, pre, post, weights):
         return dW, db
 
     @staticmethod
-    def _evolve(bias_init, eta, w_decay, w_bound, pre, post, weights, biases):
+    def _evolve(bias_init, eta, w_decay, w_bound, pre, post, preSpike,
+                postSpike, weights, biases):
         ## calculate synaptic update values
         dWeights, dBiases = GatedHebbianSynapse._compute_update(w_bound, pre, post, weights)
         weights = weights + dWeights * eta
         if bias_init != None:
             biases = biases + dBiases * eta
         if w_decay > 0.:
-            Wdec = jnp.matmul((1. - pre).T, post) * w_decay
+            Wdec = jnp.matmul((1. - preSpike).T, postSpike) * w_decay
             weights = weights - Wdec
         weights = jnp.clip(weights, 0., w_bound)
         return weights, biases, dWeights, dBiases