From 74217bfdde6f6a52a512af78ba44b62ceaa6a11f Mon Sep 17 00:00:00 2001
From: zm711 <92116279+zm711@users.noreply.github.com>
Date: Thu, 5 Oct 2023 16:44:36 -0400
Subject: [PATCH] fix cmap bug, better title for trace

---
 src/spikeanalysis/spike_plotter.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
index 90c8c8a..dd8398d 100644
--- 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,
                         )