Merci Pierre

SpikeInterface · Jun 14, 2024 · bdc7fe4 · bdc7fe4
1 parent 803463e
commit bdc7fe4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/spikeinterface/widgets/potential_merges.py b/src/spikeinterface/widgets/potential_merges.py
@@ -237,6 +237,10 @@ def _update_plot(self, change=None):
         self.w_templates._plot_probe(self.ax_probe, channel_locations, plot_unit_ids)
         crosscorrelograms_data_plot = self.w_crosscorrelograms.data_plot.copy()
         crosscorrelograms_data_plot["unit_ids"] = plot_unit_ids
+        merge_unit_indices = np.flatnonzero(np.isin(self.unique_merge_units, plot_unit_ids))
+        updated_correlograms = crosscorrelograms_data_plot["correlograms"]
+        updated_correlograms = updated_correlograms[merge_unit_indices][:, merge_unit_indices]
+        crosscorrelograms_data_plot["correlograms"] = updated_correlograms
         self.w_crosscorrelograms.plot_matplotlib(
             crosscorrelograms_data_plot, axes=self.axes_cc, ax=None, **backend_kwargs_mpl
         )