fix for infinite

zm711 · Oct 4, 2023 · 8b734c8 · 8b734c8
1 parent 5087ff6
commit 8b734c8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -798,6 +798,8 @@ def plot_response_trace(
             current_length = stim_lengths[stimulus]
             current_bins = bins[stimulus]
 
+            response[~np.isfinite(response)]=np.nan
+
             if by_trial and by_neuron:
                 for neuron in range(np.shape(response)[0]):
                     for trial in range(np.shape(response)[1]):