From 8b734c86e3a4abcaf3532d32c01fdf4e5ca7859d Mon Sep 17 00:00:00 2001
From: zm711 <92116279+zm711@users.noreply.github.com>
Date: Wed, 4 Oct 2023 17:58:33 -0400
Subject: [PATCH] fix for infinite

---
 src/spikeanalysis/spike_plotter.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
index d3f7739..99e67f9 100644
--- 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]):