[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/benchmark/benchmark_base.py

@@ -621,7 +621,6 @@ def compute_result(self):
         raise NotImplementedError
 
 
-
 # Common feature accross some benchmark : sorter + matching
 class MixinStudyUnitCount:
     def get_count_units(self, case_keys=None, well_detected_score=None, redundant_score=None, overmerged_score=None):

src/spikeinterface/benchmark/benchmark_clustering.py

@@ -195,7 +195,7 @@ def plot_performances_vs_depth_and_snr(self, *args, **kwargs):
         from .benchmark_plot_tools import plot_performances_vs_depth_and_snr
 
         return plot_performances_vs_depth_and_snr(self, *args, **kwargs)
-    
+
     def plot_performances_ordered(self, *args, **kwargs):
         from .benchmark_plot_tools import plot_performances_ordered
 

src/spikeinterface/benchmark/benchmark_plot_tools.py

@@ -14,8 +14,6 @@ def despine(ax_or_axes):
         sns.despine(ax=ax)
 
 
-
-
 def clean_axis(ax):
     for loc in ("top", "right", "left", "bottom"):
         ax.spines[loc].set_visible(False)
@@ -30,19 +28,19 @@ def plot_study_legend(study, case_keys=None, ax=None):
     Make a ax with only legend
     """
     import matplotlib.pyplot as plt
-    
+
     if case_keys is None:
         case_keys = list(study.cases.keys())
-    
+
     if ax is None:
         fig, ax = plt.subplots(figsize=figsize)
     else:
         fig = ax.get_figure()
 
     colors = study.get_colors()
-       
+
     for k in case_keys:
-        ax.plot([], color=colors[k], label=study.cases[k]['label'])
+        ax.plot([], color=colors[k], label=study.cases[k]["label"])
     ax.legend()
     clean_axis(ax)
     return fig
@@ -126,7 +124,6 @@ def plot_run_times(study, case_keys=None, levels_to_keep=None, figsize=None, ax=
         The resulting figure containing the plots
     """
     import matplotlib.pyplot as plt
-    
 
     if case_keys is None:
         case_keys = list(study.cases.keys())
@@ -138,7 +135,6 @@ def plot_run_times(study, case_keys=None, levels_to_keep=None, figsize=None, ax=
     else:
         fig = ax.get_figure()
 
-
     if levels_to_keep is None:
         colors = study.get_colors()
         labels = []
@@ -183,7 +179,6 @@ def plot_run_times(study, case_keys=None, levels_to_keep=None, figsize=None, ax=
             plt_fun = sns.barplot
             palette_keys = hues
 
-
         assert all(
             [key in colors for key in palette_keys]
         ), f"colors must have a color for each palette key: {palette_keys}"
@@ -230,7 +225,6 @@ def plot_unit_counts(study, case_keys=None, levels_to_keep=None, colors=None, fi
     if case_keys is None:
         case_keys = list(study.cases.keys())
 
-
     if ax is None:
         fig, ax = plt.subplots(figsize=figsize)
     else:
@@ -294,7 +288,6 @@ def plot_unit_counts(study, case_keys=None, levels_to_keep=None, colors=None, fi
         else:
             assert all([col in colors for col in columns]), f"colors must have a color for each column: {columns}"
 
-
         df = pd.melt(
             count_units.reset_index(),
             id_vars=levels_to_keep,
@@ -496,12 +489,11 @@ def plot_performances_vs_snr(
         if levels_to_keep is not None:
             case_group_keys, labels = study.get_grouped_keys_mapping(levels_to_group_by=levels_to_keep)
         else:
-            labels = {k: study.cases[k]['label'] for k in case_keys}
-            case_group_keys = {k : [k] for k in case_keys}
-            
+            labels = {k: study.cases[k]["label"] for k in case_keys}
+            case_group_keys = {k: [k] for k in case_keys}
 
         colors = study.get_colors(levels_to_group_by=levels_to_keep)
-        
+
         assert all([key in colors for key in case_keys]), f"colors must have a color for each case key: {case_keys}"
 
         for key, key_list in case_group_keys.items():

src/spikeinterface/benchmark/benchmark_sorter.py

@@ -65,7 +65,6 @@ def remove_benchmark(self, key):
         if sorter_folder.exists():
             shutil.rmtree(sorter_folder)
 
-
     # plotting as methods
     def plot_unit_counts(self, **kwargs):
         from .benchmark_plot_tools import plot_unit_counts