fix delay update bug (#281

fix delay update bug
brainpy · Oct 18, 2022 · daf5002 · daf5002
2 parents 87c0b86 + 7a3219a
commit daf5002
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainpy/math/delayvars.py b/brainpy/math/delayvars.py
@@ -445,8 +445,8 @@ def update(self, value: Union[float, int, bool, JaxArray, jnp.DeviceArray]):
       The value of the latest data, used to update this delay variable.
     """
     if self.update_method == ROTATION_UPDATING:
+      self.idx.value = stop_gradient((self.idx - 1) % self.num_delay_step)
       self.data[self.idx[0]] = value
-      self.idx.value = stop_gradient((self.idx + 1) % self.num_delay_step)
 
     elif self.update_method == CONCAT_UPDATING:
       if self.num_delay_step >= 2: