fix old version synapses when using brainpy.dyn.ProjAlignPost2

Author: chaoming0625

brainpy/_src/dyn/others/input.py

@@ -219,7 +219,7 @@ def __init__(
     self.reset_state(self.mode)
 
   def update(self):
-    spikes = bm.random.rand_like(self.spike) <= (self.freqs * share.dt / 1000.)
+    spikes = bm.random.rand_like(self.spike) <= (self.freqs * share['dt'] / 1000.)
     spikes = bm.asarray(spikes, dtype=self.spk_type)
     # spikes = bm.sharding.partition(spikes, self.spike.sharding)
     self.spike.value = spikes

brainpy/_src/dynold/synapses/abstract_models.py

@@ -99,7 +99,7 @@ def __init__(
       mode: bm.Mode = None,
       stop_spike_gradient: bool = False,
   ):
-    super(Delta, self).__init__(name=name,
+    super().__init__(name=name,
                                 pre=pre,
                                 post=post,
                                 conn=conn,
@@ -530,7 +530,7 @@ def __init__(
 
     self.check_post_attrs('input')
     # copy the references
-    syn = self.pre.after_updates[self.proj._syn_id].syn.syn
+    syn = self.post.before_updates[self.proj._syn_id].syn.syn
     self.g = syn.g
     self.h = syn.h
 

brainpy/_src/dynold/synapses/biological_models.py

@@ -89,7 +89,7 @@ def __init__(
                      mode=mode)
 
     # copy the references
-    syn = self.pre.after_updates[self.proj._syn_id].syn.syn
+    syn = self.post.before_updates[self.proj._syn_id].syn.syn
     self.g = syn.g
     self.spike_arrival_time = syn.spike_arrival_time
 
@@ -179,22 +179,22 @@ def __init__(
       mode: bm.Mode = None,
       stop_spike_gradient: bool = False,
   ):
-    super(GABAa, self).__init__(pre=pre,
-                                post=post,
-                                conn=conn,
-                                output=output,
-                                stp=stp,
-                                comp_method=comp_method,
-                                delay_step=delay_step,
-                                g_max=g_max,
-                                alpha=alpha,
-                                beta=beta,
-                                T=T,
-                                T_duration=T_duration,
-                                method=method,
-                                name=name,
-                                mode=mode,
-                                stop_spike_gradient=stop_spike_gradient, )
+    super().__init__(pre=pre,
+                     post=post,
+                     conn=conn,
+                     output=output,
+                     stp=stp,
+                     comp_method=comp_method,
+                     delay_step=delay_step,
+                     g_max=g_max,
+                     alpha=alpha,
+                     beta=beta,
+                     T=T,
+                     T_duration=T_duration,
+                     method=method,
+                     name=name,
+                     mode=mode,
+                     stop_spike_gradient=stop_spike_gradient, )
 
 
 class _DelayedNMDA(_DelayedSyn):
@@ -405,7 +405,7 @@ def __init__(
                      mode=mode)
 
     # copy the references
-    syn = self.pre.after_updates[self.proj._syn_id].syn.syn
+    syn = self.post.before_updates[self.proj._syn_id].syn.syn
     self.g = syn.g
     self.x = syn.x
     self.spike_arrival_time = syn.spike_arrival_time

brainpy/_src/dynold/synapses/learning_rules.py

@@ -223,7 +223,7 @@ def __init__(
                      name=name)
 
     # variables
-    syn = self.pre.after_updates[self.proj._syn_id].syn
+    syn = self.post.before_updates[self.proj._syn_id].syn.syn
     self.x = syn[0].x
     self.u = syn[0].u
     self.I = syn[1].g

brainpy/_src/dynsys.py

@@ -122,6 +122,7 @@ def __init__(
 
     # local delay variables:
     # Compatible for ``DelayRegister``
+    # TODO: will be deprecated in the future
     self.local_delay_vars: Dict = bm.node_dict()
 
     # the before- / after-updates used for computing
@@ -383,6 +384,10 @@ def update(self, *args, **kwargs):
     for node in nodes.not_subset(Dynamic).not_subset(Projection).values():
       node()
 
+    # update delays
+    # TODO: Will be deprecated in the future
+    self.update_local_delays(nodes)
+
   def reset_state(self, batch_size=None):
     nodes = self.nodes(level=1, include_self=False).subset(DynamicalSystem).unique().not_subset(DynView)
 
@@ -398,6 +403,10 @@ def reset_state(self, batch_size=None):
     for node in nodes.not_subset(Dynamic).not_subset(Projection).values():
       node.reset_state(batch_size)
 
+    # reset delays
+    # TODO: will be removed in the future
+    self.reset_local_delays(nodes)
+
   def clear_input(self):
     """Clear inputs in the children classes."""
     nodes = self.nodes(level=1, include_self=False).subset(DynamicalSystem).unique().not_subset(DynView)

brainpy/synapses.py

@@ -33,5 +33,7 @@
   DiffusiveCoupling,
   AdditiveCoupling,
 )
-
+from brainpy._src.dynold.synapses.gap_junction import (
+  GapJunction
+)