black and isort

jaxleyverse · Nov 3, 2023 · 05888b3 · 05888b3
1 parent b05aa93
commit 05888b3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/tests/neurax_vs_neuron/test_branch.py b/tests/neurax_vs_neuron/test_branch.py
@@ -63,7 +63,7 @@ def _run_neurax(i_delay, i_dur, i_amp, dt, t_max):
     branch.comp(0.0).stimulate(nx.step_current(i_delay, i_dur, i_amp, time_vec))
     branch.comp(0.0).record()
     branch.comp(1.0).record()
-    
+
     voltages = nx.integrate(branch, delta_t=dt)
 
     return voltages

diff --git a/tests/neurax_vs_neuron/test_cell.py b/tests/neurax_vs_neuron/test_cell.py
@@ -59,7 +59,7 @@ def _run_neurax(i_delay, i_dur, i_amp, dt, t_max):
     cell.branch(0).comp(0.0).record()
     cell.branch(1).comp(1.0).record()
     cell.branch(2).comp(1.0).record()
-    
+
     voltages = nx.integrate(cell, delta_t=dt)
     return voltages
 

diff --git a/tests/test_swc.py b/tests/test_swc.py
@@ -160,10 +160,12 @@ def test_swc_voltages():
     cell.set_states("h", 0.4889)
     cell.set_states("n", 0.3644787)
 
-    cell.branch(1).comp(0.05).stimulate(nx.step_current(i_delay, i_dur, i_amp, time_vec))
+    cell.branch(1).comp(0.05).stimulate(
+        nx.step_current(i_delay, i_dur, i_amp, time_vec)
+    )
     for i in trunk_inds + tuft_inds + basal_inds:
         cell.branch(i).comp(0.05).record()
-    
+
     voltages_neurax = nx.integrate(cell, delta_t=dt)
 
     ################### NEURON #################