Type annotation for offset

jaxleyverse · Nov 8, 2023 · ba3669c · ba3669c
1 parent ff4f620
commit ba3669c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/neurax/stimulus.py b/neurax/stimulus.py
@@ -8,7 +8,11 @@
 
 
 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.