Type annotation for offset

jaxleyverse · Nov 8, 2023 · 7434c39 · 7434c39
1 parent ff4f620
commit 7434c39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neurax/stimulus.py b/neurax/stimulus.py
@@ -8,7 +8,7 @@
 
 
 def step_current(
-    i_delay: float, i_dur: float, i_amp: float, time_vec: jnp.asarray, i_offset=0.0
+    i_delay: float, i_dur: float, i_amp: float, time_vec: jnp.asarray, i_offset: float = 0.0
 ):
     """
     Return step current in unit nA.