[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/spikeinterface/core/node_pipeline.py

@@ -87,10 +87,16 @@ def get_trace_margin(self):
     def get_dtype(self):
         return base_peak_dtype
 
-    def get_peak_slice(self, segment_index, start_frame, end_frame, ):
+    def get_peak_slice(
+        self,
+        segment_index,
+        start_frame,
+        end_frame,
+    ):
         # not needed for PeakDetector
         raise NotImplementedError
 
+
 # this is used in sorting components
 class PeakDetector(PeakSource):
     pass
@@ -474,7 +480,7 @@ def run_node_pipeline(
     nodes,
     job_kwargs,
     job_name="pipeline",
-    #mp_context=None,
+    # mp_context=None,
     gather_mode="memory",
     gather_kwargs={},
     squeeze_output=True,
@@ -506,7 +512,7 @@ def run_node_pipeline(
     The gather consists of concatenating features related to peaks (localization, pca, scaling, ...) into a single big vector.
     These vectors can be in "memory" or in files ("npy")
 
-    
+
     Parameters
     ----------
 
@@ -533,7 +539,7 @@ def run_node_pipeline(
     skip_after_n_peaks : None | int
         Skip the computation after n_peaks.
         This is not an exact because internally this skip is done per worker in average.
-    
+
     Returns
     -------
     outputs: tuple of np.array | np.array
@@ -596,7 +602,7 @@ def _compute_peak_pipeline_chunk(segment_index, start_frame, end_frame, worker_c
 
     recording_segment = recording._recording_segments[segment_index]
     node0 = nodes[0]
-    
+
     if isinstance(node0, (SpikeRetriever, PeakRetriever)):
         # in this case PeakSource could have no peaks and so no need to load traces just skip
         peak_slice = i0, i1 = node0.get_peak_slice(segment_index, start_frame, end_frame, max_margin)
@@ -676,7 +682,6 @@ def _compute_peak_pipeline_chunk(segment_index, start_frame, end_frame, worker_c
         return
 
 
-
 class GatherToMemory:
     """
     Gather output of nodes into list and then demultiplex and np.concatenate

src/spikeinterface/core/tests/test_node_pipeline.py

@@ -87,7 +87,7 @@ def test_run_node_pipeline(cache_folder_creation):
 
     peak_retriever = PeakRetriever(recording, peaks)
     # this test when no spikes in last chunks
-    peak_retriever_few = PeakRetriever(recording, peaks[:peaks.size//2])
+    peak_retriever_few = PeakRetriever(recording, peaks[: peaks.size // 2])
 
     # channel index is from template
     spike_retriever_T = SpikeRetriever(
@@ -212,13 +212,14 @@ def test_skip_after_n_peaks():
     nodes = [node0, node1]
 
     skip_after_n_peaks = 30
-    some_amplitudes = run_node_pipeline(recording, nodes, job_kwargs, gather_mode="memory", skip_after_n_peaks=skip_after_n_peaks)
+    some_amplitudes = run_node_pipeline(
+        recording, nodes, job_kwargs, gather_mode="memory", skip_after_n_peaks=skip_after_n_peaks
+    )
 
     assert some_amplitudes.size >= skip_after_n_peaks
     assert some_amplitudes.size < spikes.size
 
 
-
 # the following is for testing locally with python or ipython. It is not used in ci or with pytest.
 if __name__ == "__main__":
     # folder = Path("./cache_folder/core")