format: black

Author: ttngu207

element_array_ephys/ephys.py

@@ -1068,9 +1068,14 @@ def make(self, key):
             }
 
             spike_locations = sorting_analyzer.get_extension("spike_locations")
-            extremum_channel_inds = si.template_tools.get_template_extremum_channel(sorting_analyzer, outputs="index")
+            extremum_channel_inds = si.template_tools.get_template_extremum_channel(
+                sorting_analyzer, outputs="index"
+            )
             spikes_df = pd.DataFrame(
-                sorting_analyzer.sorting.to_spike_vector(extremum_channel_inds=extremum_channel_inds))
+                sorting_analyzer.sorting.to_spike_vector(
+                    extremum_channel_inds=extremum_channel_inds
+                )
+            )
 
             units = []
             for unit_idx, unit_id in enumerate(si_sorting.unit_ids):

element_array_ephys/spike_sorting/si_spike_sorting.py

@@ -114,7 +114,9 @@ def make(self, key):
             spikeglx_recording.validate_file("ap")
             data_dir = spikeglx_meta_filepath.parent
 
-            si_extractor = si.extractors.neoextractors.spikeglx.SpikeGLXRecordingExtractor
+            si_extractor = (
+                si.extractors.neoextractors.spikeglx.SpikeGLXRecordingExtractor
+            )
             stream_names, stream_ids = si.extractors.get_neo_streams(
                 "spikeglx", folder_path=data_dir
             )
@@ -125,7 +127,9 @@ def make(self, key):
             oe_probe = ephys.get_openephys_probe_data(key)
             assert len(oe_probe.recording_info["recording_files"]) == 1
             data_dir = oe_probe.recording_info["recording_files"][0]
-            si_extractor = si.extractors.neoextractors.openephys.OpenEphysBinaryRecordingExtractor
+            si_extractor = (
+                si.extractors.neoextractors.openephys.OpenEphysBinaryRecordingExtractor
+            )
 
             stream_names, stream_ids = si.extractors.get_neo_streams(
                 "openephysbinary", folder_path=data_dir