bug fix

brainpy · Dec 22, 2023 · bf6f87e · bf6f87e
1 parent 0faf6c0
commit bf6f87e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/brainpy/_src/dynold/neurons/reduced_models.py b/brainpy/_src/dynold/neurons/reduced_models.py
@@ -886,7 +886,7 @@ def __init__(
       self.integral = sdeint(method=self.method, f=self.derivative, g=self.noise)
     self.reset_state(self.mode)
 
-  def reset_state(self, batch_size=None):
+  def reset_state(self, batch_size=None, **kwargs):
     super().reset_state(batch_size)
     if self.input_var:
       self.input = variable_(bm.zeros, self.varshape, batch_size)
@@ -1023,7 +1023,7 @@ def __init__(
 
       # parameters for training
       mode: bm.Mode = None,
-      spike_fun: Callable = bm.surrogate.inv_square_grad,
+      spike_fun: Callable = bm.surrogate.inv_square_grad2,
   ):
     # initialization
     super(HindmarshRose, self).__init__(size=size,