Fix formatting

Author: nicolossus

src/expipe_plugin_cinpla/scripts/unittracking.py

@@ -8,8 +8,9 @@
 
 from expipe_plugin_cinpla.scripts.utils import _get_data_path
 from expipe_plugin_cinpla.tools.data_loader import load_spiketrains
-
-from expipe_plugin_cinpla.tools.track_units_tools import order_waveforms_by_electrode_names
+from expipe_plugin_cinpla.tools.track_units_tools import (
+    order_waveforms_by_electrode_names,
+)
 
 warnings.filterwarnings("ignore")
 

src/expipe_plugin_cinpla/tools/track_units_tools.py

@@ -18,7 +18,7 @@ def order_waveforms_by_electrode_names(templates, names):
 
     uq_names = set().union(*[set(name) for name in names]).difference({-1})
 
-    assert len(uq_names) <= templates[0].shape[1] # number of names can be max number of electrodes
+    assert len(uq_names) <= templates[0].shape[1]  # number of names can be max number of electrodes
 
     # order of the waveforms
     order = sorted(uq_names)
@@ -34,16 +34,15 @@ def order_waveforms_by_electrode_names(templates, names):
         if num_dead > 0:
             dead_index = dead_channels[0].item()
 
-
         order_indices = [np.nonzero(name == order_name)[0] for order_name in order]
         order_indices = [ind.item() if ind.size == 1 else dead_index for ind in order_indices]
 
         # in case some channel is missing from all trials
         if len(order_indices) < num_electrodes:
-            order_indices.extend([dead_index]*(num_electrodes - len(order_indices)))
+            order_indices.extend([dead_index] * (num_electrodes - len(order_indices)))
 
         action_orders.append(order_indices)
-        new_templates.append(template[:,order_indices])
+        new_templates.append(template[:, order_indices])
 
     return new_templates, action_orders
 

src/expipe_plugin_cinpla/tools/trackunitcomparison.py

@@ -15,7 +15,7 @@
     make_best_match,
     make_hungarian_match,
     make_possible_match,
-    order_waveforms_by_electrode_names
+    order_waveforms_by_electrode_names,
 )
 
 
@@ -67,15 +67,15 @@ def __init__(
 
             self.unit_ids[channel_group] = [
                 [int(u.annotations["name"]) for u in us_0],
-                [int(u.annotations["name"]) for u in us_1]
+                [int(u.annotations["name"]) for u in us_1],
             ]
             self.templates[channel_group] = [
                 [u.annotations["waveform_mean"] for u in us_0],
-                [u.annotations["waveform_mean"] for u in us_1]
+                [u.annotations["waveform_mean"] for u in us_1],
             ]
             self.electrode_names[channel_group] = [
                 [u.annotations["electrode_names"] for u in us_0],
-                [u.annotations["electrode_names"] for u in us_1]
+                [u.annotations["electrode_names"] for u in us_1],
             ]
 
             if len(us_0) > 0 and len(us_1) > 0: