fix cmap bug, better title for trace

zm711 · Oct 5, 2023 · 74217bf · 74217bf
1 parent 5febd23
commit 74217bf
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -121,6 +121,9 @@ def plot_zscores(
             indices=indices,
             show_stim=show_stim,
         )
+
+        self.cmap = None
+
         if indices:
             return sorted_cluster_ids
 
@@ -164,6 +167,8 @@ def plot_raw_firing(
         sorted_cluster_ids = self._plot_scores(
             data="raw-data", figsize=figsize, sorting_index=sorting_index, bar=bar, indices=indices, show_stim=show_stim
         )
+
+        self.cmap = None
 
         if indices:
             return sorted_cluster_ids
@@ -808,7 +813,7 @@ def plot_response_trace(
                             response[neuron, trial, :],
                             ebars=None,
                             color=color,
-                            stim=stimulus,
+                            stim=f'{stimulus}: {self.data.cluster_ids[neuron]}: {trial}',
                             show_stim=show_stim,
                             stim_lines=current_length,
                         )
@@ -822,7 +827,7 @@ def plot_response_trace(
                             avg_response,
                             ebars=ebars,
                             color=color,
-                            stim=stimulus,
+                            stim=f'{stimulus}: neuron: {self.data.cluster_ids[neuron]}',
                             show_stim=show_stim,
                             stim_lines=current_length,
                         )
@@ -832,7 +837,7 @@ def plot_response_trace(
                             avg_response,
                             ebars=None,
                             color=color,
-                            stim=stimulus,
+                            stim=f'{stimulus}: neuron: {self.data.cluster_ids[neuron]}',
                             show_stim=show_stim,
                             stim_lines=current_length,
                         )
@@ -846,7 +851,7 @@ def plot_response_trace(
                             avg_response,
                             ebars=ebars,
                             color=color,
-                            stim=stimulus,
+                            stim=f'{stimulus} event number {trial}',
                             show_stim=show_stim,
                             stim_lines=current_length,
                         )
@@ -856,7 +861,7 @@ def plot_response_trace(
                             avg_response,
                             ebars=None,
                             color=color,
-                            stim=stimulus,
+                            stim=f'{stimulus} event number {trial}',
                             show_stim=show_stim,
                             stim_lines=current_length,
                         )