fix: ensure 't' keyword argument defaults to 0 in _call_integral and format code in build method

Author: chaoming0625

brainpy/integrators/base.py

@@ -141,6 +141,10 @@ def state_delays(self, value):
         raise ValueError('Cannot set "state_delays" by users.')
 
     def _call_integral(self, *args, **kwargs):
+        kwargs = dict(kwargs)
+        t = kwargs.get('t', None)
+        kwargs['t'] = 0. if t is None else t
+
         if _during_compile:
             jaxpr, out_shapes = jax.make_jaxpr(self.integral, return_shape=True)(**kwargs)
             outs = jax.core.eval_jaxpr(jaxpr.jaxpr, jaxpr.consts, *jax.tree.leaves(kwargs))

brainpy/integrators/ode/explicit_rk.py

@@ -178,8 +178,7 @@ def __init__(self,
 
     def build(self):
         # step stage
-        common.step(self.variables, C.DT,
-                    self.A, self.C, self.code_lines, self.parameters)
+        common.step(self.variables, C.DT, self.A, self.C, self.code_lines, self.parameters)
         # variable update
         return_args = common.update(self.variables, C.DT, self.B, self.code_lines)
         # returns
@@ -189,7 +188,8 @@ def build(self):
             code_scope={k: v for k, v in self.code_scope.items()},
             code_lines=self.code_lines,
             show_code=self.show_code,
-            func_name=self.func_name)
+            func_name=self.func_name
+        )
 
 
 class Euler(ExplicitRKIntegrator):