format: black formatting

Author: ttngu207

element_array_ephys/ephys_no_curation.py

@@ -1256,7 +1256,9 @@ def make(self, key):
             unit_peak_channel = {u: chn[0] for u, chn in unit_peak_channel.items()}
 
             # reorder channel2electrode_map according to recording channel ids
-            channel_indices = sorting_analyzer.channel_ids_to_indices(sorting_analyzer.channel_ids).tolist()
+            channel_indices = sorting_analyzer.channel_ids_to_indices(
+                sorting_analyzer.channel_ids
+            ).tolist()
             channel2electrode_map = {
                 chn_idx: channel2electrode_map[chn_idx] for chn_idx in channel_indices
             }
@@ -1500,7 +1502,9 @@ def make(self, key):
         if si_sorting_analyzer_dir.exists():  # read from spikeinterface outputs
             sorting_analyzer = si.load_sorting_analyzer(folder=si_sorting_analyzer_dir)
             qc_metrics = sorting_analyzer.get_extension("quality_metrics").get_data()
-            template_metrics = sorting_analyzer.get_extension("template_metrics").get_data()
+            template_metrics = sorting_analyzer.get_extension(
+                "template_metrics"
+            ).get_data()
             metrics_df = pd.concat([qc_metrics, template_metrics], axis=1)
 
             metrics_df.rename(
@@ -1514,7 +1518,7 @@ def make(self, key):
                     "drift_mad": "cumulative_drift",
                     "half_width": "halfwidth",
                     "peak_trough_ratio": "pt_ratio",
-                    "peak_to_valley": "duration"
+                    "peak_to_valley": "duration",
                 },
                 inplace=True,
             )

element_array_ephys/spike_sorting/si_spike_sorting.py

@@ -270,28 +270,33 @@ def make(self, key):
                 overwrite=True,
             )
 
-        job_kwargs = params["SI_POSTPROCESSING_PARAMS"].get("job_kwargs", {"n_jobs": -1, "chunk_duration": "1s"})
+        job_kwargs = params["SI_POSTPROCESSING_PARAMS"].get(
+            "job_kwargs", {"n_jobs": -1, "chunk_duration": "1s"}
+        )
         extensions_params = params["SI_POSTPROCESSING_PARAMS"].get("extensions", {})
         # The order of extension computation is drawn from sorting_analyzer.get_computable_extensions()
         # each extension is parameterized by params specified in extensions_params dictionary (skip if not specified)
-        extensions_to_compute = {ext_name: extensions_params[ext_name]
-                                 for ext_name in sorting_analyzer.get_computable_extensions()
-                                 if ext_name in extensions_params}
+        extensions_to_compute = {
+            ext_name: extensions_params[ext_name]
+            for ext_name in sorting_analyzer.get_computable_extensions()
+            if ext_name in extensions_params
+        }
 
         sorting_analyzer.compute(extensions_to_compute, **job_kwargs)
 
         # Save to phy format
         if params["SI_POSTPROCESSING_PARAMS"].get("export_to_phy", False):
             si.exporters.export_to_phy(
-                sorting_analyzer=sorting_analyzer, output_folder=output_dir / sorter_name / "phy",
-                **job_kwargs
+                sorting_analyzer=sorting_analyzer,
+                output_folder=output_dir / sorter_name / "phy",
+                **job_kwargs,
             )
         # Generate spike interface report
         if params["SI_POSTPROCESSING_PARAMS"].get("export_report", True):
             si.exporters.export_report(
                 sorting_analyzer=sorting_analyzer,
                 output_folder=output_dir / sorter_name / "spikeinterface_report",
-                **job_kwargs
+                **job_kwargs,
             )
 
         self.insert1(